at master 22 lines 1.1 kB view raw
1In FHS distros rocm-core expects to be installed colocated with a full set of ROCM packages. 2In nixpkgs contexts, we don't want rocm-core to be a ROCM_PATH root. 3diff --git a/cmake_modules/rocm-core-config.cmake.in b/cmake_modules/rocm-core-config.cmake.in 4index f5fe07c..9d72c9c 100644 5--- a/cmake_modules/rocm-core-config.cmake.in 6+++ b/cmake_modules/rocm-core-config.cmake.in 7@@ -11,7 +11,14 @@ set_and_check(rocm_core_INCLUDE_DIR "@PACKAGE_INCLUDE_INSTALL_DIR@") 8 set_and_check(ROCM_CORE_INCLUDE_DIR "@PACKAGE_INCLUDE_INSTALL_DIR@") 9 set_and_check(rocm_core_LIB_DIR "@PACKAGE_LIB_INSTALL_DIR@") 10 set_and_check(ROCM_CORE_LIB_DIR "@PACKAGE_LIB_INSTALL_DIR@") 11-set_and_check(ROCM_PATH "${PACKAGE_PREFIX_DIR}") 12+# Set ROCM_PATH with priority: existing value > environment variable > package prefix 13+if(NOT DEFINED ROCM_PATH) 14+ if(DEFINED ENV{ROCM_PATH}) 15+ set(ROCM_PATH "$ENV{ROCM_PATH}") 16+ else() 17+ set_and_check(ROCM_PATH "${PACKAGE_PREFIX_DIR}") 18+ endif() 19+endif() 20 21 get_filename_component(ROCM_CORE_CMAKE_DIR "${CMAKE_CURRENT_LIST_FILE}" PATH) 22 include("${ROCM_CORE_CMAKE_DIR}/rocmCoreTargets.cmake")