Fix rccl-UnitTests packaging on Debian systems (#1846)
Signed-off-by: ravaidya <ravaidya@amd.com>
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
74d82a8145
Коммит
cbbc713b03
@@ -355,6 +355,13 @@ execute_process(
|
||||
OUTPUT_VARIABLE HOST_OS_ID
|
||||
OUTPUT_STRIP_TRAILING_WHITESPACE
|
||||
)
|
||||
|
||||
execute_process(
|
||||
COMMAND bash -c "grep '^ID_LIKE=' /etc/os-release | cut -d'=' -f2 | cut -d'\"' -f2"
|
||||
OUTPUT_VARIABLE HOST_OS_FAMILY
|
||||
OUTPUT_STRIP_TRAILING_WHITESPACE
|
||||
)
|
||||
|
||||
if (ENABLE_MSCCLPP AND NOT(${HOST_OS_ID} STREQUAL "ubuntu" OR ${HOST_OS_ID} STREQUAL "centos"))
|
||||
set(ENABLE_MSCCLPP OFF)
|
||||
message(WARNING "MSCCL++ integration not supported on this OS (${HOST_OS_ID}); disabling MSCCL++ build")
|
||||
|
||||
@@ -163,6 +163,9 @@ if(BUILD_TESTS)
|
||||
target_link_libraries(${test_executable} PRIVATE ${RCCL_COMMON_LINK_LIBS})
|
||||
if(BUILD_SHARED_LIBS)
|
||||
target_link_libraries(${test_executable} PRIVATE rccl)
|
||||
if(${HOST_OS_FAMILY} STREQUAL "debian")
|
||||
set_property(TARGET ${test_executable} PROPERTY INSTALL_RPATH "${CMAKE_BINARY_DIR}")
|
||||
endif()
|
||||
else()
|
||||
add_dependencies(${test_executable} rccl)
|
||||
target_link_libraries(${test_executable} PRIVATE dl rt numa -lrccl -L${CMAKE_BINARY_DIR} -lrocm_smi64 -L${ROCM_PATH}/lib -L${ROCM_PATH}/rocm_smi/lib)
|
||||
|
||||
Ссылка в новой задаче
Block a user