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 
[ROCm/rccl commit: 35024ca1cb]
This commit is contained in:
Arm Patinyasakdikul
2025-06-27 15:29:37 -05:00
committed by GitHub
parent 23e5680038
commit 4d71cae249
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/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));