Streamlined setting up MPI in the CMakeLists

2020-08-19 13:44:35 +03:00
parent b035dfd6fb
commit f9e9a4a4ef
2 changed files with 2 additions and 12 deletions
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -78,6 +78,8 @@ if (DOUBLE_PRECISION)
    add_definitions(-DAC_DOUBLE_PRECISION=1)
 endif ()
 if (MPI_ENABLED)
+    find_package(MPI REQUIRED)
+    include_directories(${MPI_CXX_INCLUDE_DIRS})
    add_definitions(-DAC_MPI_ENABLED=1)
 endif ()
 if (MULTIGPU_ENABLED)
@@ -92,11 +94,6 @@ include_directories(src/common)                                # Common headers
 include_directories(${CMAKE_BINARY_DIR})                       # DSL headers
 include_directories(${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}) # CUDA headers

-if (MPI_ENABLED)
-    find_package(MPI REQUIRED)
-    include_directories(${MPI_CXX_INCLUDE_DIRS})
-endif()
-
 ## Subdirectories
 add_subdirectory(src/utils)
 add_subdirectory(src/core/kernels)