From 4cd86b185c16aea8ed6ebccf1e9574b36300e714 Mon Sep 17 00:00:00 2001 From: akolliasAMD <99202231+akolliasAMD@users.noreply.github.com> Date: Fri, 3 Nov 2023 09:10:52 -0600 Subject: [PATCH] MSCCL stream fix (#948) [ROCm/rccl commit: 988efe605a84ce694168b78edc09cf1f56ac7a64] --- projects/rccl/src/misc/msccl/msccl_setup.cc | 1 - 1 file changed, 1 deletion(-) diff --git a/projects/rccl/src/misc/msccl/msccl_setup.cc b/projects/rccl/src/misc/msccl/msccl_setup.cc index 0b6c997ef2..5222cf39c6 100644 --- a/projects/rccl/src/misc/msccl/msccl_setup.cc +++ b/projects/rccl/src/misc/msccl/msccl_setup.cc @@ -67,7 +67,6 @@ ncclResult_t mscclSetupScratch(struct mscclAlgo* hostAlgo, hipStream_t stream) { mscclStatus& status = mscclGetStatus(); size_t sizeNeeded = (status.nBytes * (size_t)(hostAlgo->nScratchChunks)) / (size_t)(hostAlgo->nChunksPerLoop); if (sizeNeeded > status.scratchBufferSize){ - CUDACHECK(hipStreamSynchronize(stream)); NCCLCHECK(ncclCudaFree(status.scratchBuffer)); NCCLCHECK(ncclCudaCalloc((char**)&status.scratchBuffer, sizeNeeded)); status.scratchBufferSize = sizeNeeded;