ROCM: Fix GPU MPI operations support in build (horovod#3746)

Signed-off-by: R0n12 <xulang7@gmail.com>
chongxiaoc · Oct 17, 2022 · 44f9357 · 44f9357
1 parent 80e4894
commit 44f9357
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -209,6 +209,9 @@ if(DEFINED HOROVOD_GPU_ALLREDUCE OR DEFINED HOROVOD_GPU_ALLGATHER OR DEFINED HOR
         set(CMAKE_CXX_FLAGS "${ROCM_COMPILE_FLAGS} ${CMAKE_CXX_FLAGS}")
         list(APPEND SOURCES "${PROJECT_SOURCE_DIR}/horovod/common/ops/hip_operations.cc"
                             "${PROJECT_SOURCE_DIR}/horovod/common/ops/gpu_operations.cc")
+        if(HAVE_MPI)
+            list(APPEND SOURCES "${PROJECT_SOURCE_DIR}/horovod/common/ops/mpi_gpu_operations.cc")
+        endif()
         add_definitions(-DHAVE_ROCM=1 -DHAVE_GPU=1)
         set(HAVE_ROCM TRUE)
     else()