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
2021-06-09 13:24:26 -07:00
committed by GitHub
vanhempi c2064adcc7
commit b815a2800f
18 muutettua tiedostoa jossa 79 lisäystä ja 36 poistoa
+3
Näytä tiedosto
@@ -98,6 +98,8 @@ struct ncclComm {
uint64_t opCount;
// Collective operation counter
uint64_t collOpCount;
// P2P operation counter
uint64_t p2pOpCount;
// Channels for collectives
int nChannels;
@@ -127,6 +129,7 @@ struct ncclComm {
// Flags for enable P2P NET
uint32_t *p2pNet;
uint32_t useIntraNet;
// Device side of the communicator
struct ncclDevComm *devComm;