From d1da885fafd55236fa934220289118da8e6053b0 Mon Sep 17 00:00:00 2001
From: xu98bin <78574951+xu98bin@users.noreply.github.com>
Date: Fri, 3 Feb 2023 11:47:50 +0800
Subject: [PATCH] solve auto_aprallel pp2 with fp16 question (#49913)

---
 python/paddle/distributed/auto_parallel/completion.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/python/paddle/distributed/auto_parallel/completion.py b/python/paddle/distributed/auto_parallel/completion.py
index 8960c47c1f..8979239df5 100644
--- a/python/paddle/distributed/auto_parallel/completion.py
+++ b/python/paddle/distributed/auto_parallel/completion.py
@@ -1850,11 +1850,11 @@ class Completer:
                             op_dist_attr.set_output_dims_mapping(
                                 input_var.name, ref_dims_mapping
                             )
-
-                        input_var_attr.process_mesh = ref_process_mesh
-                        self._dist_context.set_tensor_dist_attr_for_program(
-                            input_var, input_var_attr
-                        )
+                        if "SkipUpdate" not in input_name:
+                            input_var_attr.process_mesh = ref_process_mesh
+                            self._dist_context.set_tensor_dist_attr_for_program(
+                                input_var, input_var_attr
+                            )
 
                     self._dist_context.set_op_dist_attr_for_program(
                         op, op_dist_attr
-- 
GitLab