MSCCL: Fix memcpy bug (#721)

This commit is contained in:
Ziyue Yang authored and GitHub committed 2023-04-11 14:46:53 -07:00
1 orang tua dc149a9fbd
melakukan 7289c05146
1 file changed
+1 -1
+1 -1
Melihat File
@@ -143,7 +143,7 @@ __device__ __forceinline__ void mscclRunInterpreter(
// initialize mscclShmem.mscclTB
threadBlockCopy(
(uint64_t *)&mscclShmem.mscclTB, (uint64_t *)(algo->mscclTBs + bid),
sizeof(struct mscclThreadBlock), tid, nthreads);
sizeof(struct mscclThreadBlock) / sizeof(uint64_t), tid, nthreads);
__synclds(); // publish mscclShmem.mscclTB.channelId
// initialize ncclShmem and mscclShmem.work