yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Explanation of Function change

Browse Source
This commit is contained in:
WoosungMyung 2025-01-04 08:47:46 +09:00
parent 2ea4ee94bf
commit 1875119095
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/bootstrap.cc
View File

@ -70,7 +70,7 @@ static int localIdFromRoot(int rank, int root, int nRanks, int nRoots) {
int ir = BOOTSTRAP_PID(root, nRoots); int ir = BOOTSTRAP_PID(root, nRoots);
return rank - firstRankFromRoot(ir, nRanks, nRoots); return rank - firstRankFromRoot(ir, nRanks, nRoots);
} }
// return the number of child for a root, root will be periodized // Check if the given rank is the first rank from the root
static int isFirstFromRoot(int rank, int root, int nRanks, int nRoots) { static int isFirstFromRoot(int rank, int root, int nRanks, int nRoots) {
return (rank == firstRankFromRoot(root, nRanks, nRoots)); return (rank == firstRankFromRoot(root, nRanks, nRoots));
} }