add include and lib

CaRoLZhangxy · Jul 1, 2024 · 4e30cd4 · 4e30cd4
1 parent 504bd29
commit 4e30cd4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/source/op/pt/CMakeLists.txt b/source/op/pt/CMakeLists.txt
@@ -21,6 +21,8 @@ endif()
 find_package(MPI)
 if(MPI_FOUND)
   include(CheckCXXSymbolExists)
+  set(CMAKE_REQUIRED_INCLUDES ${MPI_INCLUDE_PATH})
+  set(CMAKE_REQUIRED_LIBRARIES ${MPI_LIBRARIES})
   check_cxx_symbol_exists(MPIX_Query_cuda_support "${MPI_INCLUDE_PATH}/mpi.h"
                           CUDA_AWARE)
   if(NOT CUDA_AWARE)