Fix typo in ncclGetKernelIndex macro (#1424)

This commit is contained in:
Bertan Dogancay
2024-11-18 10:40:05 -05:00
committed by GitHub
parent 4336a0f3a3
commit dfe4a3ed81
2 changed files with 6 additions and 3 deletions
+1 -1
View File
@@ -31,7 +31,7 @@ struct ncclKernelMatch {
};
#ifdef ENABLE_COLLTRACE
#define ncclGetKernelIndex(p_comm) ((p_comm)->unroll + (p_comm)->collTraceThread ? 2 : 0)
#define ncclGetKernelIndex(p_comm) ((p_comm)->unroll + ((p_comm)->collTraceThread ? 2 : 0))
static ncclKernelMatch const ncclKerns[4] = {
{(void *)ncclDevKernel_Generic, true},
{(void *)ncclDevKernel_Generic_4, true},