set MAXCHANNELS to 128

[ROCm/rccl commit: ef442f8f92]
This commit is contained in:
Nusrat Islam
2024-05-23 15:27:43 -05:00
والد 2447161fb4
کامیت 48821ad0d7
7فایلهای تغییر یافته به همراه33 افزوده شده و 12 حذف شده
@@ -241,7 +241,7 @@ __forceinline__ __device__ void ncclKernelMain(struct ncclDevComm* comm, struct
y = __popcll(channelMask.masks[i] & ((1ull<<x)-1));
y = total + y;
if (blockIdx.x == y) {
ncclShmem.channelId = y;
ncclShmem.channelId = x + total;
break;
}
}
@@ -251,7 +251,7 @@ __forceinline__ __device__ void ncclKernelMain(struct ncclDevComm* comm, struct
y = __popcll(channelMask.masks[i] & ((1ull<<x)-1));
y = y + total;
if (blockIdx.x == y) {
ncclShmem.channelId = y;
ncclShmem.channelId = x + total;
break;
}
}