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
This commit is contained in:
Wenkai Du authored and GitHub committed 2024-02-21 17:15:11 -08:00
1 parent b275ed0b56
commit c5ab37211b
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/init.cc
+1 -1
View File
@@ -443,7 +443,7 @@ NCCL_PARAM(AggChannelSize, "AGG_CHANNEL_SIZE", -2);
NCCL_PARAM(DisableGraphHelper, "GRAPH_HELPER_DISABLE", 0);
// GDRCOPY support: FIFO_ENABLE when enabled locates a workFifo in CUDA memory
NCCL_PARAM(GdrCopyFifoEnable, "GDRCOPY_FIFO_ENABLE", 1);
NCCL_PARAM(WorkFifoDepth, "WORK_FIFO_DEPTH", 64<<10);
NCCL_PARAM(WorkFifoDepth, "WORK_FIFO_DEPTH", 256<<10);
enum ncclLaunchMode ncclParamLaunchMode;