Setup collectives threshold for enabling intranet (#387)

* Setup collectives threshold for enabling intranet

* Use separate operation counters for coll and p2p
This commit is contained in:
Wenkai Du authored and GitHub committed 2021-06-09 13:24:26 -07:00
1 parent c2064adcc7
commit b815a2800f
18 files changed
+79 -36

No files matched your search

+1 -3
View File
@@ -1017,11 +1017,9 @@ ncclResult_t ncclTopoGetNetDev(struct ncclTopoSystem* system, int rank, struct n
return ncclSuccess;
}
extern int64_t rcclParamP2pNetDisable();
ncclResult_t ncclTopoGetIntraNetDev(struct ncclTopoSystem* system, int rank, struct ncclTopoGraph* graph, int channelId, int type, int* dev) {
*dev = -1;
if (graph && graph->nIntraChannels && rcclParamP2pNetDisable() == 0) {
if (graph && graph->nIntraChannels) {
int n1 = -1;
int ngpus = system->nodes[GPU].count;
int nnets = system->nodes[NET].count;