Enable collnet in RCCL (#333)

* Enable CollNet and use different number of channels

* topo_expl: enable collnet

[ROCm/rccl commit: 1d6244b18d]
此提交包含在:
Wenkai Du
2021-03-19 12:58:13 -07:00
提交者 GitHub
父節點 260b54adb9
當前提交 287ed0f18a
共有 14 個檔案被更改,包括 393 行新增24 行删除
+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];