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

Query XGMI link count through rocm_smi_lib API (#442)

Browse Source
This commit is contained in:
Wenkai Du
2021-10-26 10:30:20 -07:00
committed by GitHub
parent d23dfc12c1
commit 14a184eb67
4 changed files with 67 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/include/rocm_smi_wrap.h
+2 -1
View File
@@ -24,10 +24,11 @@ THE SOFTWARE.
#define ROCM_SMI_WRAP_H_
#include "rocm_smi/rocm_smi.h"
#include "rocm_smi/rocm_smi64Config.h"
#include "nccl.h"
ncclResult_t rocm_smi_init();
ncclResult_t rocm_smi_getDeviceIndexByPciBusId(const char* pciBusId, uint32_t* deviceIndex);
ncclResult_t rocm_smi_getLinkInfo(int srcDev, int dstDev, RSMI_IO_LINK_TYPE* rsmi_type, int *hops, int *bw);
ncclResult_t rocm_smi_getLinkInfo(int srcDev, int dstDev, RSMI_IO_LINK_TYPE* rsmi_type, int *hops, int *count);
#endif