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

make binary tree work on 2.13.4

Browse Source 
[ROCm/rccl commit: 8f3219dbd4]
This commit is contained in:
Edgar Gabriel committed 2022-09-15 00:01:54 +00:00
1 parent 069af6f7c3
commit 95d6ed2154
3 files changed
+10 -7
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rccl/src/enqueue.cc
+1 -1
View File
@@ -1156,7 +1156,7 @@ static ncclResult_t getAlgoInfo(struct ncclInfo* info, int collNetTypeSupport, i
if (info->nBytes <= 2200008) {
info->protocol = NCCL_PROTO_LL;
info->algorithm = NCCL_ALGO_TREE;
info->nChannels = 24;
info->nChannels = std::min(24, comm->nChannels);
} else {
info->protocol = NCCL_PROTO_SIMPLE;
info->algorithm = NCCL_ALGO_RING;