Enable local sendrecv over network if GDR is available on all GPUs (#324)

[ROCm/rccl commit: c018edf0f2]
このコミットが含まれているのは:
Wenkai Du
2021-03-05 19:59:41 -08:00
committed by GitHub
コミット bcf4ecb0e3
14個のファイルの変更163行の追加44行の削除
+5
ファイルの表示
@@ -150,6 +150,8 @@ struct ncclTree {
struct ncclPeer {
struct ncclConnector send;
struct ncclConnector recv;
struct ncclConnector p2pSend;
struct ncclConnector p2pRecv;
};
struct ncclDevComm;
@@ -332,6 +334,9 @@ struct ncclDevComm {
// Channels, device side
struct ncclChannel* channels;
// Flags for enable P2P NET
uint32_t *p2pNet;
#ifdef ENABLE_PROFILING
// Profiling counters
struct ncclProf* devProf;