add contiguous in grad output

deepmodeling · Jun 27, 2024 · d18c432 · d18c432
1 parent 17cdcb0
commit d18c432
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/op/pt/comm.cc b/source/op/pt/comm.cc
@@ -197,7 +197,7 @@ class Border : public torch::autograd::Function<Border> {
     torch::Tensor nlocal_tensor = saved_variables[6];
     torch::Tensor nghost_tensor = saved_variables[7];
 
-    torch::Tensor d_local_g1_tensor = grad_output[0];
+    torch::Tensor d_local_g1_tensor = grad_output[0].contiguous();
 #ifdef USE_MPI
     int mpi_init = 0;
     MPI_Initialized(&mpi_init);