diff --git a/CMakeLists.txt b/CMakeLists.txt index 9de9ad0211..b209ebfcf1 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -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") diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt index 86347ff36a..98878b1e76 100644 --- a/test/CMakeLists.txt +++ b/test/CMakeLists.txt @@ -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)