[ROCM] Fix undefined symbols by adding library (#8446)

* Add libhsa-runtime64 reference. * Remove lib in library definition.
apache · Jul 12, 2021 · 1d7a9e9 · 1d7a9e9
1 parent 3424005
commit 1d7a9e9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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)