Skip to content

Commit 6f1d5d2

Browse files
committed
fix hip build
1 parent 110b505 commit 6f1d5d2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ggml-cuda.cu

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6683,9 +6683,9 @@ static void * ggml_cuda_pool_malloc_vmm(size_t size, size_t * actual_size) {
66836683
g_cuda_pool_handles[id].push_back(handle);
66846684
g_cuda_pool_size[id] += reserve_size;
66856685

6686-
//printf("cuda pool[%d]: size increased to %llu MB (reserved %llu MB)\n",
6687-
// id, (unsigned long long) (g_cuda_pool_size[id]/1024/1024),
6688-
// (unsigned long long) (reserve_size/1024/1024));
6686+
printf("cuda pool[%d]: size increased to %llu MB (reserved %llu MB)\n",
6687+
id, (unsigned long long) (g_cuda_pool_size[id]/1024/1024),
6688+
(unsigned long long) (reserve_size/1024/1024));
66896689
}
66906690

66916691
GGML_ASSERT(g_cuda_pool_addr[id] != 0);
@@ -6835,7 +6835,7 @@ void ggml_init_cublas() {
68356835
g_tensor_split[id] = total_vram;
68366836
total_vram += prop.totalGlobalMem;
68376837
#if defined(GGML_USE_HIPBLAS) && defined(__HIP_PLATFORM_AMD__)
6838-
g_device_caps[id] = 100*prop.major + 10*prop.minor + CC_OFFSET_AMD;
6838+
g_device_caps[id].cc = 100*prop.major + 10*prop.minor + CC_OFFSET_AMD;
68396839
#else
68406840
g_device_caps[id].cc = 100*prop.major + 10*prop.minor;
68416841
#endif // defined(GGML_USE_HIPBLAS) && defined(__HIP_PLATFORM_AMD__)

0 commit comments

Comments
 (0)