Check rocm_smi64Config.h on older ROCm build (#452)

Cette révision appartient à :
Wenkai Du
2021-10-28 07:26:28 -07:00
révisé par GitHub
Parent d221fb672a
révision 4643a17f83
3 fichiers modifiés avec 8 ajouts et 1 suppressions
+2
Voir le fichier
@@ -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();
+1 -1
Voir le fichier
@@ -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));