diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt index 0b17b0202b..4c8ebf73c5 100644 --- a/test/CMakeLists.txt +++ b/test/CMakeLists.txt @@ -53,9 +53,9 @@ if(BUILD_TESTS) # through -l and -L instead of command line input. if(BUILD_STATIC) add_dependencies(UnitTests rccl) - target_link_libraries(UnitTests PRIVATE dl rt numa -lrccl -L${CMAKE_BINARY_DIR}) + target_link_libraries(UnitTests PRIVATE dl rt numa -lrccl -L${CMAKE_BINARY_DIR} -lrocm_smi64 -L${ROCM_PATH}/rocm_smi/lib) add_dependencies(UnitTestsMultiProcess rccl) - target_link_libraries(UnitTestsMultiProcess PRIVATE dl rt numa -lrccl -L${CMAKE_BINARY_DIR}) + target_link_libraries(UnitTestsMultiProcess PRIVATE dl rt numa -lrccl -L${CMAKE_BINARY_DIR} -lrocm_smi64 -L${ROCM_PATH}/rocm_smi/lib) else() target_link_libraries(UnitTests PRIVATE rccl) target_link_libraries(UnitTestsMultiProcess PRIVATE rt rccl)