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

2021-10-28 07:26:28 -07:00
parent d221fb672a
commit 4643a17f83
3 changed files with 8 additions and 1 deletions
@@ -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();