Fix problem of transpose InferSPMD in backward. (#60049)

GhostScreaming · web-flow · commit 8189a9908603 · 2023-12-17T18:21:37.000+08:00
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}};
 }

Original file line number	Diff line number	Diff line change
`@@ -197,7 +197,6 @@ SpmdInfo TransposeGradInferSpmd(const DistMetaTensor& out_grad,`
`197`	`197`	`x_dims_mapping[origin_index] = out_grad_dims_mapping[i];`
`198`	`198`	`}`
`199`	`199`	`TensorDistAttr x_grad_dist_attr = out_grad.dist_attr();`
`200`		`- x_grad_dist_attr.clean_partial_status();`
`201`	`200`	`x_grad_dist_attr.set_dims_mapping(x_dims_mapping);`
`202`	`201`	`return {{out_grad.dist_attr()}, {x_grad_dist_attr}};`
`203`	`202`	`}`