AI/rocm-systems
2
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[topo-expl] update header file location. (#1769)

Browse Source
This commit is contained in:
Arm Patinyasakdikul authored and GitHub committed 2025-06-27 15:29:37 -05:00
1 parent 16101e654f
commit 35024ca1cb
2 files changed
+8 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tools/topo_expl/model.cpp
+5 -3
View File
@@ -191,7 +191,7 @@ struct setupReq {
int remoteRank;
int shared;
int netDev;
int useGdr;
enum ncclTopoGdrMode useGdr;
int channelId;
int connIndex;
};
@@ -263,7 +263,8 @@ ncclResult_t collNetCanConnect(int* ret, struct ncclComm* comm, struct ncclTopo
}
ncclResult_t collNetSendSetup(struct ncclComm* comm, struct ncclTopoGraph* graph, struct ncclPeerInfo* myInfo, struct ncclPeerInfo* peerInfo, struct ncclConnect* connectInfo, struct ncclConnector* send, int channelId, int connIndex) {
int netDev, useGdr = 0, proxy;
int netDev, proxy;
enum ncclTopoGdrMode useGdr = ncclTopoGdrModeDisable;
NCCLCHECK(ncclTopoGetNetDev(comm, myInfo->rank, graph, channelId, peerInfo->rank, NULL, &netDev, &proxy));
NCCLCHECK(ncclTopoCheckGdr(comm->topo, myInfo->rank, netDev, 1, &useGdr));
@@ -273,7 +274,8 @@ ncclResult_t collNetSendSetup(struct ncclComm* comm, struct ncclTopoGraph* graph
}
ncclResult_t collNetRecvSetup(struct ncclComm* comm, struct ncclTopoGraph* graph, struct ncclPeerInfo* myInfo, struct ncclPeerInfo* peerInfo, struct ncclConnect* connectInfo, struct ncclConnector* recv, int channelId, int connIndex) {
int netDev, useGdr = 0, proxy;
int netDev, proxy;
enum ncclTopoGdrMode useGdr = ncclTopoGdrModeDisable;
NCCLCHECK(ncclTopoGetNetDev(comm, myInfo->rank, graph, channelId, peerInfo->rank, NULL, &netDev, &proxy));
NCCLCHECK(ncclTopoCheckGdr(comm->topo, myInfo->rank, netDev, 0, &useGdr));