[ROCm/rccl commit: 988efe605a]
Этот коммит содержится в:
akolliasAMD
2023-11-03 09:10:52 -06:00
коммит произвёл GitHub
родитель 297023a7a6
Коммит 4cd86b185c
-1
Просмотреть файл
@@ -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;