Enable collnet in RCCL (#333)
* Enable CollNet and use different number of channels
* topo_expl: enable collnet
[ROCm/rccl commit: 1d6244b18d]
Этот коммит содержится в:
1 родитель
260b54adb9
Коммит
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