From 5fd1ad7db616de5d81213a83ff4dd4c5a7d20c5d Mon Sep 17 00:00:00 2001 From: Ghost Screaming Date: Fri, 15 Dec 2023 14:54:00 +0800 Subject: [PATCH] Fix problem of transpose InferSPMD in backward. (#60000) --- paddle/phi/infermeta/spmd_rules/transpose.cc | 1 - 1 file changed, 1 deletion(-) diff --git a/paddle/phi/infermeta/spmd_rules/transpose.cc b/paddle/phi/infermeta/spmd_rules/transpose.cc index 95840f8edfe35..82fc0390a4614 100644 --- 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}}; }