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

CMakeLists.txt

@@ -213,6 +213,11 @@ if (BUILD_ALLREDUCE_ONLY)
   add_definitions(-DBUILD_ALLREDUCE_ONLY)
 endif()
 
+CHECK_INCLUDE_FILE_CXX("${ROCM_PATH}/rocm_smi/include/rocm_smi/rocm_smi64Config.h" HAVE_ROCM_SMI64CONFIG)
+IF(HAVE_ROCM_SMI64CONFIG)
+  add_definitions(-DUSE_ROCM_SMI64CONFIG)
+ENDIF()
+
 foreach(target ${AMDGPU_TARGETS})
   target_link_libraries(rccl PRIVATE --amdgpu-target=${target})
 endforeach()

src/include/rocm_smi_wrap.h

@@ -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();

src/misc/rocm_smi_wrap.cc

@@ -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));