Fix problem of transpose InferSPMD in backward. (PaddlePaddle#60000)

RichardWooSJTU · Dec 15, 2023 · 5fd1ad7 · 5fd1ad7
1 parent 257f5d3
commit 5fd1ad7
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/paddle/phi/infermeta/spmd_rules/transpose.cc b/paddle/phi/infermeta/spmd_rules/transpose.cc
@@ -197,7 +197,6 @@ SpmdInfo TransposeGradInferSpmd(const DistMetaTensor& out_grad,
     x_dims_mapping[origin_index] = out_grad_dims_mapping[i];
   }
   TensorDistAttr x_grad_dist_attr = out_grad.dist_attr();
-  x_grad_dist_attr.clean_partial_status();
   x_grad_dist_attr.set_dims_mapping(x_dims_mapping);
   return {{out_grad.dist_attr()}, {x_grad_dist_attr}};
 }