diff --git a/cmake/modules/ROCM.cmake b/cmake/modules/ROCM.cmake index b908df2f869b..8b064d2eb2eb 100644 --- a/cmake/modules/ROCM.cmake +++ b/cmake/modules/ROCM.cmake @@ -34,6 +34,9 @@ if(USE_ROCM) file(GLOB RUNTIME_ROCM_SRCS src/runtime/rocm/*.cc) list(APPEND RUNTIME_SRCS ${RUNTIME_ROCM_SRCS}) list(APPEND TVM_RUNTIME_LINKER_LIBS ${ROCM_HIPHCC_LIBRARY}) + if (ROCM_HSA_LIBRARY) + list(APPEND TVM_RUNTIME_LINKER_LIBS ${ROCM_HSA_LIBRARY}) + endif() if(USE_MIOPEN) message(STATUS "Build with MIOpen support") diff --git a/cmake/utils/FindROCM.cmake b/cmake/utils/FindROCM.cmake index 7d4e282956d9..4d895ff89d13 100644 --- a/cmake/utils/FindROCM.cmake +++ b/cmake/utils/FindROCM.cmake @@ -55,6 +55,7 @@ macro(find_rocm use_rocm) endif() find_library(ROCM_MIOPEN_LIBRARY MIOpen ${__rocm_sdk}/lib) find_library(ROCM_ROCBLAS_LIBRARY rocblas ${__rocm_sdk}/lib) + find_library(ROCM_HSA_LIBRARY hsa-runtime64 ${__rocm_sdk}/lib) if(ROCM_HIPHCC_LIBRARY) set(ROCM_FOUND TRUE)