Fix(critical): Introduce SMI_DISABLED define to forcibly disable SMI usage in headers

This commit is contained in:
Donato Capitella
2026-02-01 10:39:37 +00:00
parent cd91b85935
commit 2e6df33acc
3 changed files with 3 additions and 2 deletions
+1
View File
@@ -284,6 +284,7 @@ if(ROCM_VERSION VERSION_GREATER_EQUAL "71100" AND ENABLE_AMDSMI)
else()
# Force unset if disabled
unset(USE_AMDSMI CACHE)
add_definitions("-DSMI_DISABLED")
endif()
if(NOT USE_AMDSMI AND ENABLE_AMDSMI)
+1 -1
View File
@@ -7,7 +7,7 @@
#include "amd_smi/amdsmi.h"
#include "nccl.h"
#ifdef USE_AMDSMI
#if defined(USE_AMDSMI) && !defined(SMI_DISABLED)
ncclResult_t amd_smi_init();
ncclResult_t amd_smi_shutdown();
ncclResult_t amd_smi_getNumDevice(uint32_t* num_devs);
+1 -1
View File
@@ -29,7 +29,7 @@ THE SOFTWARE.
#endif
#include "nccl.h"
#ifdef USE_ROCMSMI
#if defined(USE_ROCMSMI) && !defined(SMI_DISABLED)
ncclResult_t rocm_smi_init();
ncclResult_t rocm_smi_getNumDevice(uint32_t* num_devs);
ncclResult_t rocm_smi_getDevicePciBusIdString(uint32_t deviceIndex, char* pciBusId, size_t len);