From 835f7ebf377ae55e69af089fc13ec568d7c658ff Mon Sep 17 00:00:00 2001 From: Mike Vermeulen Date: Wed, 23 Jun 2021 09:21:23 -0500 Subject: [PATCH 1/2] Add libhsa-runtime64 reference. --- cmake/modules/ROCM.cmake | 3 +++ cmake/utils/FindROCM.cmake | 1 + 2 files changed, 4 insertions(+) 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..b2300b76bd62 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 libhsa-runtime64 ${__rocm_sdk}/lib) if(ROCM_HIPHCC_LIBRARY) set(ROCM_FOUND TRUE) From fa98e37368789ffd4b26de14dd35dcb3610531cc Mon Sep 17 00:00:00 2001 From: Mike Vermeulen Date: Wed, 23 Jun 2021 10:51:21 -0500 Subject: [PATCH 2/2] Remove lib in library definition. --- cmake/utils/FindROCM.cmake | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmake/utils/FindROCM.cmake b/cmake/utils/FindROCM.cmake index b2300b76bd62..4d895ff89d13 100644 --- a/cmake/utils/FindROCM.cmake +++ b/cmake/utils/FindROCM.cmake @@ -55,7 +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 libhsa-runtime64 ${__rocm_sdk}/lib) + find_library(ROCM_HSA_LIBRARY hsa-runtime64 ${__rocm_sdk}/lib) if(ROCM_HIPHCC_LIBRARY) set(ROCM_FOUND TRUE)