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

Merge pull request #1563 from LameloBally/comment

Browse Source 
Comment of a Bootstrap Function  changed
This commit is contained in:
David Addison 2025-01-03 16:04:33 -08:00 committed by GitHub
commit 8fb057cda2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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));
} }