Skip to content

Commit 48022e8

Browse files
author
Bodhi Hu
committed
add comment for the MUSA .cc calculation block
1 parent 981adc1 commit 48022e8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ggml/src/ggml-cuda/ggml-cuda.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -262,6 +262,7 @@ static ggml_cuda_device_info ggml_cuda_init() {
262262
id, prop.name, prop.gcnArchName, info.devices[id].cc & 0xffff,
263263
device_vmm ? "yes" : "no", prop.warpSize);
264264
#elif defined(GGML_USE_MUSA)
265+
// TODO: refine the .cc to reflect MUSA's actual CC capabilities
265266
info.devices[id].smpbo = prop.sharedMemPerBlockOptin;
266267
info.devices[id].cc = 100*prop.major + 10*prop.minor;
267268
GGML_LOG_INFO(" Device %d: %s, compute capability %d.%d, VMM: %s\n",

0 commit comments

Comments
 (0)