Set ring cuda msallreduce as default

Tixxx · Aug 23, 2019 · a1093e2 · a1093e2
1 parent 84288ad
commit a1093e2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/horovod/common/operations.cc b/horovod/common/operations.cc
@@ -49,6 +49,7 @@
 
 #if HAVE_CUDA
 #include "ops/msallreduce_cuda_operations.h"
+#include "ops/msallreduce_cuda_ring_operations.h"
 #include "ops/cuda_operations.h"
 #include "ops/mpi_cuda_operations.h"
 #endif
@@ -157,7 +158,7 @@ OperationManager* CreateOperationManager(HorovodGlobalState& state) {
 #if HOROVOD_GPU_ALLREDUCE == 'M'
   if (state.msallreduce_enabled == true){
     LOG(INFO) << "msallGpureduce enabled.";
-    msallreduce_ops.push_back(std::shared_ptr<AllreduceOp>(new MsCudaAllreduceOp(&mpi_context, &cuda_context, &state)));
+    msallreduce_ops.push_back(std::shared_ptr<AllreduceOp>(new MsCudaRingAllreduceOp(&mpi_context, &cuda_context, &state)));
   }
   allreduce_ops.push_back(std::shared_ptr<AllreduceOp>(
       new MPI_CUDAAllreduce(&mpi_context, &cuda_context, &state)));