modify include path(still not found)

CaRoLZhangxy · Jul 1, 2024 · b8cce46 · b8cce46
1 parent 9cc67a7
commit b8cce46
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/source/op/pt/CMakeLists.txt b/source/op/pt/CMakeLists.txt
@@ -21,7 +21,7 @@ endif()
 find_package(MPI)
 if(MPI_FOUND)
   include(CheckCXXSymbolExists)
-  check_cxx_symbol_exists(MPIX_Query_cuda_support "mpi.h" CUDA_AWARE)
+  check_cxx_symbol_exists(MPIX_Query_cuda_support "${MPI_INCLUDE_PATH}/mpi.h" CUDA_AWARE)
   if(NOT CUDA_AWARE)
     target_compile_definitions(deepmd_op_pt PRIVATE NO_CUDA_AWARE)
   endif()

diff --git a/source/op/pt/comm.cc b/source/op/pt/comm.cc
@@ -102,7 +102,7 @@ class Border : public torch::autograd::Function<Border> {
       if (version >= 4) {
 #ifdef NO_CUDA_AWARE
         cuda_aware = 0;
-#elif
+#else
         cuda_aware = MPIX_Query_cuda_support();
 #endif
       } else {