updates for cudart support

ollama · Mar 22, 2024 · b4cb7f2 · b4cb7f2
1 parent 43ae1ed
commit b4cb7f2
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 39 deletions.
diff --git a/gpu/gpu_info_cudart.c b/gpu/gpu_info_cudart.c
@@ -124,40 +124,6 @@ void cudart_check_vram(cudart_handle_t h, mem_info_t *resp) {
       return;
     }
 
-    if (h.verbose) {
-      cudartBrandType_t brand = 0;
-      // When in verbose mode, report more information about
-      // the card we discover, but don't fail on error
-      // Need to map out alternatives of these for CUDART libraries
-      // For now just returning a generic "unsupported" error.
-      ret = CUDART_UNSUPPORTED;
-      if (ret != CUDART_SUCCESS) {
-        LOG(h.verbose, "nvmlDeviceGetName unsupported with CUDART libraries: %d\n", ret);
-      } else {
-        LOG(h.verbose, "[%d] CUDA device name: %s\n", i, buf);
-      }
-      if (ret != CUDART_SUCCESS) {
-        LOG(h.verbose, "nvmlDeviceGetBoardPartNumber unsupported with CUDART libraries: %d\n", ret);
-      } else {
-        LOG(h.verbose, "[%d] CUDA part number: %s\n", i, buf);
-      }
-      if (ret != CUDART_SUCCESS) {
-        LOG(h.verbose, "nvmlDeviceGetSerial unsupported with CUDART libraries: %d\n", ret);
-      } else {
-        LOG(h.verbose, "[%d] CUDA S/N: %s\n", i, buf);
-      }
-      if (ret != CUDART_SUCCESS) {
-        LOG(h.verbose, "nvmlDeviceGetVbiosVersion unsupported with CUDART libraries: %d\n", ret);
-      } else {
-        LOG(h.verbose, "[%d] CUDA vbios version: %s\n", i, buf);
-      }
-      if (ret != CUDART_SUCCESS) {
-        LOG(h.verbose, "nvmlDeviceGetBrand unsupported with CUDART libraries: %d\n", ret);
-      } else {
-        LOG(h.verbose, "[%d] CUDA brand: %d\n", i, brand);
-      }
-    }
-
     LOG(h.verbose, "[%d] CUDA totalMem %lu\n", i, memInfo.total);
     LOG(h.verbose, "[%d] CUDA freeMem %lu\n", i, memInfo.free);
 

diff --git a/gpu/gpu_info_cudart.h b/gpu/gpu_info_cudart.h
@@ -27,11 +27,6 @@ typedef struct cudartDriverVersion {
   int minor;
 } cudartDriverVersion_t;
 
-typedef enum cudartBrandType_enum
-{
-    CUDART_BRAND_UNKNOWN          = 0,
-} cudartBrandType_t;
-
 typedef struct cudart_handle {
   void *handle;
   uint16_t verbose;