Setup collectives threshold for enabling intranet (#387)

* Setup collectives threshold for enabling intranet

* Use separate operation counters for coll and p2p
Этот коммит содержится в:
Wenkai Du
2021-06-09 13:24:26 -07:00
коммит произвёл GitHub
родитель c2064adcc7
Коммит b815a2800f
18 изменённых файлов: 79 добавлений и 36 удалений
+3
Просмотреть файл
@@ -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;