Enable collnet in RCCL (#333)

* Enable CollNet and use different number of channels

* topo_expl: enable collnet

[ROCm/rccl commit: 1d6244b18d]
This commit is contained in:
Wenkai Du authored and GitHub committed 2021-03-19 12:58:13 -07:00
1 parent 260b54adb9
commit 287ed0f18a
14 files changed
+393 -24

No files matched your search

+2
View File
@@ -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];