diff --git a/CMakeLists.txt b/CMakeLists.txt index 52b392a13..32edeea25 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -545,12 +545,15 @@ if (LLAMA_VULKAN) endif() if (LLAMA_HIPBLAS) - if ($ENV{ROCM_PATH}) - set(ROCM_PATH $ENV{ROCM_PATH}) - else() - set(ROCM_PATH /opt/rocm) + if (NOT EXISTS $ENV{ROCM_PATH}) + if (NOT EXISTS /opt/rocm) + set(ROCM_PATH /usr) + else() + set(ROCM_PATH /opt/rocm) + endif() endif() list(APPEND CMAKE_PREFIX_PATH ${ROCM_PATH}) + list(APPEND CMAKE_PREFIX_PATH "$ENV{ROCM_PATH}/lib64/cmake") # CMake on Windows doesn't support the HIP language yet if(WIN32)