From eef4cf6a4c3f880275bf951c46aae09a607f4549 Mon Sep 17 00:00:00 2001 From: Daya Khudia Date: Wed, 17 May 2023 23:48:37 +0000 Subject: [PATCH] fix multi-gpu build --- 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..22f735c27 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 -lmpi -lmpi_cxx logger) endif() add_library(nccl_utils STATIC nccl_utils.cc)