diff --git a/mlir/tools/mlir-rocm-runner/CMakeLists.txt b/mlir/tools/mlir-rocm-runner/CMakeLists.txt index 9b07d00d80961..2c0791d7a5c1d 100644 --- a/mlir/tools/mlir-rocm-runner/CMakeLists.txt +++ b/mlir/tools/mlir-rocm-runner/CMakeLists.txt @@ -38,7 +38,7 @@ if(MLIR_ROCM_RUNNER_ENABLED) add_definitions(-D__ROCM_PATH__="${ROCM_PATH}") # Locate HIP runtime library. - find_library(ROCM_RUNTIME_LIBRARY hip_hcc + find_library(ROCM_RUNTIME_LIBRARY amdhip64 PATHS "${HIP_PATH}/lib") if (NOT ROCM_RUNTIME_LIBRARY) message(SEND_ERROR "Could not locate ROCm HIP runtime library")