AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

set MAXCHANNELS to 128

Browse Source
This commit is contained in:
Nusrat Islam
2024-05-23 15:27:43 -05:00
parent 9f654f6cf5
commit ef442f8f92
7 changed files with 33 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/device/common.h
+2 -2
View File
@@ -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;
}
}