From ec3ad67f0bc3935a42790685e65e19fb7ef55999 Mon Sep 17 00:00:00 2001 From: Liangfu Chen Date: Tue, 16 May 2023 15:38:13 -0700 Subject: [PATCH 1/2] Update CMakeLists.txt --- CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 870e67f0a..9e81b0a9f 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -418,7 +418,7 @@ add_library(transformer-shared SHARED if (BUILD_MULTI_GPU) target_link_libraries(transformer-shared PUBLIC - -lmpi + ${MPI_mpi_LIBRARY} ${NCCL_LIBRARIES} ) endif() From 8e74e8ee4ff089bc972d10bc0a95581731272faf Mon Sep 17 00:00:00 2001 From: Liangfu Chen Date: Tue, 16 May 2023 15:39:23 -0700 Subject: [PATCH 2/2] Update CMakeLists.txt --- src/fastertransformer/utils/CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/fastertransformer/utils/CMakeLists.txt b/src/fastertransformer/utils/CMakeLists.txt index 9796ad076..b88045a53 100644 --- a/src/fastertransformer/utils/CMakeLists.txt +++ b/src/fastertransformer/utils/CMakeLists.txt @@ -57,7 +57,7 @@ add_library(mpi_utils STATIC mpi_utils.cc) set_property(TARGET mpi_utils PROPERTY POSITION_INDEPENDENT_CODE ON) set_property(TARGET mpi_utils PROPERTY CUDA_RESOLVE_DEVICE_SYMBOLS ON) if (BUILD_MULTI_GPU) - target_link_libraries(mpi_utils PUBLIC -lmpi logger) + target_link_libraries(mpi_utils PUBLIC ${MPI_mpi_LIBRARY} logger) endif() add_library(nccl_utils STATIC nccl_utils.cc)