fix hip build

ggerganov · Dec 23, 2023 · 6f1d5d2 · 6f1d5d2
1 parent 110b505
commit 6f1d5d2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ggml-cuda.cu b/ggml-cuda.cu
@@ -6683,9 +6683,9 @@ static void * ggml_cuda_pool_malloc_vmm(size_t size, size_t * actual_size) {
         g_cuda_pool_handles[id].push_back(handle);
         g_cuda_pool_size[id] += reserve_size;
 
-        //printf("cuda pool[%d]: size increased to %llu MB (reserved %llu MB)\n",
-        //       id, (unsigned long long) (g_cuda_pool_size[id]/1024/1024),
-        //       (unsigned long long) (reserve_size/1024/1024));
+        printf("cuda pool[%d]: size increased to %llu MB (reserved %llu MB)\n",
+               id, (unsigned long long) (g_cuda_pool_size[id]/1024/1024),
+               (unsigned long long) (reserve_size/1024/1024));
     }
 
     GGML_ASSERT(g_cuda_pool_addr[id] != 0);
@@ -6835,7 +6835,7 @@ void ggml_init_cublas() {
             g_tensor_split[id] = total_vram;
             total_vram += prop.totalGlobalMem;
 #if defined(GGML_USE_HIPBLAS) && defined(__HIP_PLATFORM_AMD__)
-            g_device_caps[id] = 100*prop.major + 10*prop.minor + CC_OFFSET_AMD;
+            g_device_caps[id].cc = 100*prop.major + 10*prop.minor + CC_OFFSET_AMD;
 #else
             g_device_caps[id].cc = 100*prop.major + 10*prop.minor;
 #endif // defined(GGML_USE_HIPBLAS) && defined(__HIP_PLATFORM_AMD__)