From 270aa41f6b5cba2aeb086dbe3a1cf080334d8036 Mon Sep 17 00:00:00 2001 From: "Wen-Heng (Jack) Chung" Date: Tue, 7 Nov 2023 13:47:24 -0600 Subject: [PATCH] Use send instead of sendWithBarrier. (#727) [ROCm/rccl commit: efc42d9045a71ca1e6f05cf0803d5c89e663cd36] --- projects/rccl/src/collectives/device/msccl_kernel_impl.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/projects/rccl/src/collectives/device/msccl_kernel_impl.h b/projects/rccl/src/collectives/device/msccl_kernel_impl.h index d2c8adb016..31ca61ad1c 100644 --- a/projects/rccl/src/collectives/device/msccl_kernel_impl.h +++ b/projects/rccl/src/collectives/device/msccl_kernel_impl.h @@ -312,7 +312,7 @@ __device__ __forceinline__ void mscclRunInterpreter( NpKit::CollectGpuEventLDS(NPKIT_EVENT_MSCCL_SEND_ENTRY, thisNelem*sizeof(T), 0, NPKIT_GET_GPU_TIMESTAMP()); } #endif - prims.sendWithBarrier(srcOffset, thisNelem); // LL.send is the only situation where there is no barrier at the end. + prims.send(srcOffset, thisNelem); // LL.send is the only situation where there is no barrier at the end. #if defined(ENABLE_NPKIT) && defined(ENABLE_NPKIT_EVENT_MSCCL_SEND_EXIT) if (tid == 0) {