-
Notifications
You must be signed in to change notification settings - Fork 1.1k
Open
Description
Lines 252 to 271 in 25368a7
| while (true) { | |
| if (t == nullptr) { | |
| teamIsNew = true; | |
| t = (struct ncclDevrTeam*)malloc(sizeof(struct ncclDevrTeam) + team.nRanks*sizeof(int)); | |
| t->team = team; | |
| t->mcHandle = 0x0; | |
| t->mcBasePtr = nullptr; | |
| for (int i=0; i < team.nRanks; i++) { | |
| t->worldRankList[i] = comm->rank + (i - team.rank)*team.stride; | |
| } | |
| break; | |
| } else if (t->team.rank == team.rank && t->team.nRanks == team.nRanks && t->team.stride == team.stride) { | |
| if (!multimem || t->mcBasePtr != nullptr) { | |
| // Matching team is sufficient | |
| if (outTeam) *outTeam = t; | |
| return ncclSuccess; | |
| } | |
| break; // Need to enable multimem | |
| } | |
| } |
NCCL Version
2.29.3
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels