diff --git a/discover/gpu_info.h b/discover/gpu_info.h index 094b791a8..ee7ff4c33 100644 --- a/discover/gpu_info.h +++ b/discover/gpu_info.h @@ -27,12 +27,14 @@ #endif +#ifndef LOG #define LOG(verbose, ...) \ do { \ if (verbose) { \ fprintf(stderr, __VA_ARGS__); \ } \ } while (0) +#endif #ifdef __cplusplus extern "C" { diff --git a/discover/gpu_info_cudart.c b/discover/gpu_info_cudart.c index 79f19f27f..bc5115bfd 100644 --- a/discover/gpu_info_cudart.c +++ b/discover/gpu_info_cudart.c @@ -1,6 +1,7 @@ #ifndef __APPLE__ // TODO - maybe consider nvidia support on intel macs? #include +#include #include "gpu_info_cudart.h" void cudart_init(char *cudart_lib_path, cudart_init_resp_t *resp) { @@ -168,9 +169,9 @@ void cudart_bootstrap(cudart_handle_t h, int i, mem_info_t *resp) { resp->free = memInfo.free; resp->used = memInfo.used; - LOG(h.verbose, "[%s] CUDA totalMem %llu\n", resp->gpu_id, resp->total); - LOG(h.verbose, "[%s] CUDA freeMem %llu\n", resp->gpu_id, resp->free); - LOG(h.verbose, "[%s] CUDA usedMem %llu\n", resp->gpu_id, resp->used); + LOG(h.verbose, "[%s] CUDA totalMem %" PRId64 "\n", resp->gpu_id, resp->total); + LOG(h.verbose, "[%s] CUDA freeMem %" PRId64 "\n", resp->gpu_id, resp->free); + LOG(h.verbose, "[%s] CUDA usedMem %" PRId64 "\n", resp->gpu_id, resp->used); LOG(h.verbose, "[%s] Compute Capability %d.%d\n", resp->gpu_id, resp->major, resp->minor); } @@ -180,4 +181,4 @@ void cudart_release(cudart_handle_t h) { h.handle = NULL; } -#endif // __APPLE__ \ No newline at end of file +#endif // __APPLE__ diff --git a/discover/gpu_info_nvcuda.c b/discover/gpu_info_nvcuda.c index 636cba91c..d2d0b683b 100644 --- a/discover/gpu_info_nvcuda.c +++ b/discover/gpu_info_nvcuda.c @@ -1,6 +1,7 @@ #ifndef __APPLE__ // TODO - maybe consider nvidia support on intel macs? #include +#include #include "gpu_info_nvcuda.h" void nvcuda_init(char *nvcuda_lib_path, nvcuda_init_resp_t *resp) { @@ -193,8 +194,8 @@ void nvcuda_bootstrap(nvcuda_handle_t h, int i, mem_info_t *resp) { resp->total = memInfo.total; resp->free = memInfo.free; - LOG(h.verbose, "[%s] CUDA totalMem %llu mb\n", resp->gpu_id, resp->total / 1024 / 1024); - LOG(h.verbose, "[%s] CUDA freeMem %llu mb\n", resp->gpu_id, resp->free / 1024 / 1024); + LOG(h.verbose, "[%s] CUDA totalMem %" PRId64 "mb\n", resp->gpu_id, resp->total / 1024 / 1024); + LOG(h.verbose, "[%s] CUDA freeMem %" PRId64 "mb\n", resp->gpu_id, resp->free / 1024 / 1024); LOG(h.verbose, "[%s] Compute Capability %d.%d\n", resp->gpu_id, resp->major, resp->minor); @@ -247,4 +248,4 @@ void nvcuda_release(nvcuda_handle_t h) { h.handle = NULL; } -#endif // __APPLE__ \ No newline at end of file +#endif // __APPLE__