AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[MSCCL]: Move scratch buffer debug msgs to TRACE (#1189)

Browse Source 
Signed-off-by: nileshnegi <Nilesh.Negi@amd.com>

[ROCm/rccl commit: 1249a6c3fd]
This commit is contained in:
Nilesh M Negi
2024-05-31 17:54:23 -05:00
committed by GitHub
parent b35e2b8c4b
commit 8518ef4afb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/misc/msccl/msccl_setup.cc
+1 -1
View File
@@ -458,7 +458,7 @@ ncclResult_t mscclSetupKernel(const void* sendBuff, void* recvBuff, size_t count
NCCLCHECK(ncclCudaMalloc((char**)&scratchBuffer, sizeRounded, hipDeviceMallocFinegrained));
#endif
work.scratchBuffer = status.scratchBuffers[sizeRounded] = scratchBuffer;
INFO(NCCL_INIT, "MSCCL: Allocated scratch buffer of size %lu on request (%lu)", sizeRounded, sizeNeeded);
TRACE(NCCL_INIT, "MSCCL: Allocated scratch buffer of size %lu on request (%lu)", sizeRounded, sizeNeeded);
} else {
work.scratchBuffer = itr->second;
}