diff --git a/CMakeLists.txt b/CMakeLists.txt index 4757c07667..5d9989ea3a 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -7,8 +7,6 @@ set( CMAKE_CXX_STANDARD 14 ) # Without this line, it will add -std=gnu++14 instead, which has some issues. set( CMAKE_CXX_EXTENSIONS OFF ) -set(CMAKE_INSTALL_PREFIX "${ROCM_PATH}" CACHE PATH "") - project(rccl CXX) include(cmake/Dependencies.cmake) @@ -23,6 +21,14 @@ if( CMAKE_CXX_COMPILER MATCHES ".*/hipcc$" ) string(REGEX MATCH ".mcode\-object\-version" TARGET_ID_SUPPORT ${CXX_OUTPUT}) endif() +if(NOT DEFINED ROCM_PATH) + get_filename_component(_real_path ${CMAKE_CXX_COMPILER} REALPATH) + get_filename_component(_new_path "${_real_path}" DIRECTORY) + get_filename_component(ROCM_PATH "${_new_path}/../.." REALPATH) +endif() + +set(CMAKE_INSTALL_PREFIX "${ROCM_PATH}" CACHE PATH "") + #Set the AMDGPU_TARGETS with backward compatiblity if(COMMAND rocm_check_target_ids) rocm_check_target_ids(DEFAULT_AMDGPU_TARGETS