Этот коммит содержится в:
Wenkai Du
2023-12-01 16:01:11 -08:00
коммит произвёл GitHub
родитель 0b53f79196
Коммит 4ba65d1d6a
4 изменённых файлов: 4 добавлений и 11 удалений
+2 -2
Просмотреть файл
@@ -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;