MSCCL: Fix memcpy bug (#721)

This commit is contained in:
Ziyue Yang
2023-04-12 05:46:53 +08:00
committato da GitHub
parent dc149a9fbd
commit 7289c05146
+1 -1
Vedi 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