From 7289c05146eec635db5399e32d72efc1b96db138 Mon Sep 17 00:00:00 2001 From: Ziyue Yang Date: Wed, 12 Apr 2023 05:46:53 +0800 Subject: [PATCH] MSCCL: Fix memcpy bug (#721) --- src/collectives/device/msccl_kernel.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/collectives/device/msccl_kernel.cu b/src/collectives/device/msccl_kernel.cu index c90d6b3c55..0948d193f3 100644 --- a/src/collectives/device/msccl_kernel.cu +++ b/src/collectives/device/msccl_kernel.cu @@ -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