OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Update CMakeLists for rocm

Browse Source
This commit is contained in:
Tong Dong Qiu
2023-01-25 14:15:47 +01:00
parent 11ea024fbe
commit eaef4b8c85
4 changed files with 47 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
CMakeLists.txt
View File

@@ -209,12 +209,47 @@ else()
endif()
endif()
find_package(rocblas)
find_package(rocsparse)
message(INFO " HAVE_ROCSPARSE: ${HAVE_ROCSPARSE}")
find_package(hip)
if(rocsparse_FOUND AND rocblas_FOUND)
set(HAVE_ROCSPARSE 1)
set(COMPILE_BDA_BRIDGE 1)
message(INFO " hip version: ${hip_VERSION}")
message(INFO " rocblas version: ${rocblas_VERSION}")
message(INFO " rocsparse version: ${rocsparse_VERSION}")
message(INFO " CMAKE_CXX_COMPILER: ${CMAKE_CXX_COMPILER}")
message(INFO " HAVE_ROCSPARSE: ${HAVE_ROCSPARSE}")
if(hip_FOUND)
#message(INFO " CXX_COMPILER: ${CXX_COMPILER}")
get_filename_component(CXX_COMPILER ${CMAKE_CXX_COMPILER} NAME)
#message(INFO " CXX_COMPILER: ${CXX_COMPILER}")
message(INFO " CMAKE_CXX_COMPILER_ID: ${CMAKE_CXX_COMPILER_ID}")
if(hip_VERSION VERSION_LESS "5.3")
if(CMAKE_CXX_COMPILER_ID STREQUAL "GNU")
message(WARNING " Cannot use gcc for rocm < 5.3")
unset(HAVE_ROCSPARSE)
endif()
find_package(rocalution)
if(ROCALUTION_FOUND)
set(HAVE_ROCALUTION 1)
set(COMPILE_BDA_BRIDGE 1)
endif()
else()
find_package(rocblas)
find_package(rocsparse)
if(rocsparse_FOUND AND rocblas_FOUND)
set(HAVE_ROCSPARSE 1)
set(COMPILE_BDA_BRIDGE 1)
endif()
if(${CXX_COMPILER} STREQUAL "hipcc")
find_package(rocalution)
if(ROCALUTION_FOUND)
set(HAVE_ROCALUTION 1)
set(COMPILE_BDA_BRIDGE 1)
endif()
else()
message(WARNING " Cannot use hipcc for rocm >= 5.4")
endif()
endif()
endif()
find_package(amgcl)
@@ -248,12 +283,6 @@ if(OpenCL_FOUND)
endif()
endif()
find_package(rocalution)
if(ROCALUTION_FOUND)
set(HAVE_ROCALUTION 1)
set(COMPILE_BDA_BRIDGE 1)
endif()
macro (config_hook)
opm_need_version_of ("dune-common")