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:
1 parent
260b54adb9
commit
287ed0f18a
14 files changed
+393
-24
No files matched your search
@@ -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];
|
||||
|
||||
Reference in new issue
Block a user