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

Update RCCL/MSCCL work FIFO depth to 256K (#1091)

Browse Source 
[ROCm/rccl commit: c5ab37211b]
This commit is contained in:
Wenkai Du
2024-02-21 17:15:11 -08:00
committed by GitHub
parent 4b4bdd904e
commit 4b31894d70
2 changed files with 2 additions and 2 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
@@ -20,7 +20,7 @@
RCCL_PARAM(MscclEnableDoneEvent, "MSCCL_ENABLE_DONE_EVENT", 1);
#endif
RCCL_PARAM(MscclWorkFifoDepth, "MSCCL_WORK_FIFO_DEPTH", 64<<10);
RCCL_PARAM(MscclWorkFifoDepth, "MSCCL_WORK_FIFO_DEPTH", 256<<10);
static inline size_t computeSizeNeeded(size_t nBytes, int nScratchChunks, int nChunksPerLoop) {
return (nBytes * (size_t)nScratchChunks) / (size_t)nChunksPerLoop;