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

Simple tree changes (#599)

Browse Source 
changed treebase to create basic balanced tree

[ROCm/rccl commit: 3c1b1ec8c8]
This commit is contained in:
akolliasAMD authored and GitHub committed 2022-08-19 13:51:49 -06:00
1 parent 27cb7d2b20
commit 1d55fe756c
2 files changed
+30 -69
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rccl/src/enqueue.cc
+2 -5
View File
@@ -464,13 +464,10 @@ static ncclResult_t getAlgoInfo(struct ncclInfo* info, int collNetTypeSupport, i
info->nChannels = nc;
if (!userTuneInput) {
// always respect user settings
if (info->nBytes <= 900000) {
if (info->nBytes <= 2200008) {
info->protocol = NCCL_PROTO_LL;
info->algorithm = NCCL_ALGO_TREE;
info->nChannels = std::min(comm->nChannels, info->nBytes <= 16384? 4 : 24);
} else if (info->nBytes <= 2200008) {
info->protocol = NCCL_PROTO_LL;
info->algorithm = NCCL_ALGO_RING;
info->nChannels = 24;
} else {
info->protocol = NCCL_PROTO_SIMPLE;
info->algorithm = NCCL_ALGO_RING;