Increase max channles to 64 (#993)

2023-12-01 16:01:11 -08:00
parent 0b53f79196
commit 4ba65d1d6a
4 changed files with 4 additions and 11 deletions
@@ -476,8 +476,8 @@ __forceinline__ __device__ void ncclKernel(
      int y = __popcll(channelMask & ((1ull<<x)-1));
      if (blockIdx.x == y) ncclShmem.channelId = x;
    }
-    if (32 < MAXCHANNELS) {
-      x = 32 + tid;
+    if (WARP_SIZE < MAXCHANNELS) {
+      x = WARP_SIZE + tid;
      if (channelMask & (1ull<<x)) {
        int y = __popcll(channelMask & ((1ull<<x)-1));
        if (blockIdx.x == y) ncclShmem.channelId = x;