Check rocm_smi64Config.h on older ROCm build (#452)
Cette révision appartient à :
@@ -24,7 +24,9 @@ THE SOFTWARE.
|
||||
#define ROCM_SMI_WRAP_H_
|
||||
|
||||
#include "rocm_smi/rocm_smi.h"
|
||||
#ifdef USE_ROCM_SMI64CONFIG
|
||||
#include "rocm_smi/rocm_smi64Config.h"
|
||||
#endif
|
||||
#include "nccl.h"
|
||||
|
||||
ncclResult_t rocm_smi_init();
|
||||
|
||||
@@ -90,7 +90,7 @@ ncclResult_t rocm_smi_getLinkInfo(int srcDev, int dstDev, RSMI_IO_LINK_TYPE* rsm
|
||||
*count = 1;
|
||||
if (*rsmi_type == RSMI_IOLINK_TYPE_XGMI && rsmi_weight == 15) {
|
||||
*hops = 1;
|
||||
#if rocm_smi_VERSION_MAJOR >= 5
|
||||
#if defined USE_ROCM_SMI64CONFIG && rocm_smi_VERSION_MAJOR >= 5
|
||||
uint64_t min_bw = 0, max_bw = 0;
|
||||
rsmi_version_t version;
|
||||
ROCMSMICHECK(rsmi_version_get(&version));
|
||||
|
||||
Référencer dans un nouveau ticket
Bloquer un utilisateur