Unchange tensor in neighbor allreduce

Bluefog-Lib · Jun 12, 2020 · ff22ae2 · ff22ae2
1 parent 144bbab
commit ff22ae2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bluefog/torch/mpi_ops.cc b/bluefog/torch/mpi_ops.cc
@@ -350,7 +350,7 @@ int DoNeighborAllreduce(::torch::Tensor tensor, ::torch::Tensor output,
               }
             }
             output.resize_(shape_vector);
-            output.add_(tensor.mul_(self_weight));
+            output.add_(tensor.mul(self_weight));
           } else {
             auto output_reduced = output.slice(0, 0, first_dim);
             for (int i = 1; i < bluefog_neighbor_size(); i++) {
@@ -418,7 +418,7 @@ int DoNeighborAllreduce(::torch::Tensor tensor, ::torch::Tensor output,
               }
             }
             output.resize_(shape_vector);
-            output.add_(tensor.mul_(self_weight));
+            output.add_(tensor.mul(self_weight));
           } else {
             auto output_reduced = output.slice(0, 0, first_dim);
             for (int i = 1; i < bluefog_neighbor_size(); i++) {