* Enable CollNet and use different number of channels

* topo_expl: enable collnet

[ROCm/rccl commit: 1d6244b18d]
Этот коммит содержится в:
Wenkai Du authored and GitHub committed 2021-03-19 12:58:13 -07:00
1 родитель 260b54adb9
Коммит 287ed0f18a
14 files changed
+393 -24

No files matched your search

+2
Просмотреть файл
@@ -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];