Nilesh M Negi
7ca67f1cb9
Signed-off-by: nileshnegi <Nilesh.Negi@amd.com>
[ROCm/rccl commit: 5aaf7121d9]
20 خطوط
581 B
CMake
20 خطوط
581 B
CMake
|
|
if (DEFINED ENV{ROCM_PATH})
|
|
set(rocm_bin "$ENV{ROCM_PATH}/bin")
|
|
else()
|
|
set(ROCM_PATH "/opt/rocm" CACHE PATH "Path to the ROCm installation.")
|
|
set(rocm_bin "/opt/rocm/bin")
|
|
endif()
|
|
|
|
if (NOT DEFINED ENV{CXX})
|
|
set(CMAKE_CXX_COMPILER "${rocm_bin}/hipcc" CACHE PATH "Path to the C++ compiler")
|
|
else()
|
|
set(CMAKE_CXX_COMPILER "$ENV{CXX}" CACHE PATH "Path to the C++ compiler")
|
|
endif()
|
|
|
|
if (NOT DEFINED ENV{CC})
|
|
set(CMAKE_C_COMPILER "${rocm_bin}/hipcc" CACHE PATH "Path to the C compiler")
|
|
else()
|
|
set(CMAKE_C_COMPILER "$ENV{CC}" CACHE PATH "Path to the C compiler")
|
|
endif()
|