Enable collnet in RCCL (#333)

* Enable CollNet and use different number of channels

* topo_expl: enable collnet
Bu işleme şunda yer alıyor:
Wenkai Du
2021-03-19 12:58:13 -07:00
işlemeyi yapan: GitHub
ebeveyn 244e25d980
işleme 1d6244b18d
14 değiştirilmiş dosya ile 393 ekleme ve 24 silme
+2
Dosyayı Görüntüle
@@ -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];