MSCCL: Fix memcpy bug (#721)

此提交包含在:
Ziyue Yang authored and GitHub committed 2023-04-11 14:46:53 -07:00
1 父節點 dc149a9fbd
當前提交 7289c05146
1 file changed
+1 -1
+1 -1
查看文件
@@ -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