[CMAKE] Compatible for ROCm before 3.7 (apache#6359)

neo-ai · Sep 18, 2020 · 4eb2744 · 4eb2744
1 parent 637717c
commit 4eb2744
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmake/util/FindROCM.cmake b/cmake/util/FindROCM.cmake
@@ -49,8 +49,13 @@ macro(find_rocm use_rocm)
   if(__rocm_sdk)
     set(ROCM_INCLUDE_DIRS ${__rocm_sdk}/include)
     find_library(ROCM_HIPHCC_LIBRARY amdhip64 ${__rocm_sdk}/lib)
+    # Backward compatible with before ROCm3.7
+    if(NOT ROCM_HIPHCC_LIBRARY)
+      find_library(ROCM_HIPHCC_LIBRARY hip_hcc ${__rocm_sdk}/lib)
+    endif()
     find_library(ROCM_MIOPEN_LIBRARY MIOpen ${__rocm_sdk}/lib)
     find_library(ROCM_ROCBLAS_LIBRARY rocblas ${__rocm_sdk}/lib)
+
     if(ROCM_HIPHCC_LIBRARY)
       set(ROCM_FOUND TRUE)
     endif()