Setup collectives threshold for enabling intranet (#387)

* Setup collectives threshold for enabling intranet

* Use separate operation counters for coll and p2p

[ROCm/rccl commit: b815a2800f]
This commit is contained in:
Wenkai Du
2021-06-09 13:24:26 -07:00
committed by GitHub
parent ab9b9151d2
commit 5bebcb0015
18 changed files with 79 additions and 36 deletions
+3
View File
@@ -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;