[BUILD] Update install.sh for RCCL build (#1191)
Signed-off-by: nileshnegi <Nilesh.Negi@amd.com>
This commit is contained in:
gecommit door
GitHub
bovenliggende
1249a6c3fd
commit
5aaf7121d9
@@ -0,0 +1,19 @@
|
||||
|
||||
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()
|
||||
Verwijs in nieuw issue
Block a user