Query XGMI link count through rocm_smi_lib API (#442)

[ROCm/rccl commit: 14a184eb67]
이 커밋은 다음에 포함됨:
Wenkai Du 작성하고 GitHub가 2021-10-26 10:30:20 -07:00를 커밋함
1 부모 0afd607328
커밋 15143b1cfb
4 파일이 변경됨
+67 -56

검색과 일치하는 파일 없음

+2 -1
파일 보기
@@ -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