diff --git a/mlir/lib/ExecutionEngine/CMakeLists.txt b/mlir/lib/ExecutionEngine/CMakeLists.txt --- a/mlir/lib/ExecutionEngine/CMakeLists.txt +++ b/mlir/lib/ExecutionEngine/CMakeLists.txt @@ -262,11 +262,15 @@ set(CMAKE_PREFIX_PATH "${REAL_CMAKE_PREFIX_PATH}") if (NOT DEFINED ROCM_TEST_CHIPSET) - execute_process(COMMAND "${ROCM_PATH}/bin/rocm_agent_enumerator" - OUTPUT_VARIABLE AGENTS_STRING - ERROR_VARIABLE AGENTS_STRING - RESULT_VARIABLE AGENT_ENUMERATOR_RESULT) - + find_program(ROCM_AGENT_ENUMERATOR rocm_agent_enumerator "${ROCM_PATH}/bin" /usr/bin /usr/local/bin) + if(ROCM_AGENT_ENUMERATOR) + execute_process(COMMAND "${ROCM_AGENT_ENUMERATOR}" + OUTPUT_VARIABLE AGENTS_STRING + ERROR_VARIABLE AGENTS_STRING + RESULT_VARIABLE AGENT_ENUMERATOR_RESULT) + else() + message(SEND_ERROR "Could not find rocm_agent_enumerator") + endif() if (NOT AGENT_ENUMERATOR_RESULT EQUAL 0) message(SEND_ERROR "Could not run rocm_agent_enumerator and ROCM_TEST_CHIPSET is not defined") set(AGENTS_STRING "")