Fix rccl-UnitTests packaging on Debian systems (#1846)

Signed-off-by: ravaidya <ravaidya@amd.com>
Этот коммит содержится в:
Rahul Vaidya
2025-08-08 12:28:56 -05:00
коммит произвёл GitHub
родитель 74d82a8145
Коммит cbbc713b03
2 изменённых файлов: 10 добавлений и 0 удалений
+7
Просмотреть файл
@@ -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")
+3
Просмотреть файл
@@ -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)