Enable collnet in RCCL (#333)

* Enable CollNet and use different number of channels

* topo_expl: enable collnet
Dieser Commit ist enthalten in:
Wenkai Du
2021-03-19 12:58:13 -07:00
committet von GitHub
Ursprung 244e25d980
Commit 1d6244b18d
14 geänderte Dateien mit 393 neuen und 24 gelöschten Zeilen
+2
Datei anzeigen
@@ -101,6 +101,8 @@ struct ncclComm {
int p2pnChannels;
int p2pnChannelsPerPeer;
int p2pChannels[MAXCHANNELS];
//Channels for collnet
int collNetnChannels;
// Buffer sizes
int buffSizes[NCCL_NUM_PROTOCOLS];