solve auto_aprallel pp2 with fp16 question (#49913)

d1da885f · xu98bin · GitHub · 6b151c0e · d1da885f
隐藏空白更改
内联并排

Showing with 5 addition and 5 deletion

python/paddle/distributed/auto_parallel/completion.py python/paddle/distributed/auto_parallel/completion.py +5 -5

未找到文件。
--- 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
                            )
+                        if "SkipUpdate" not in input_name:
-                        input_var_attr.process_mesh = ref_process_mesh
+                            input_var_attr.process_mesh = ref_process_mesh
-                        self._dist_context.set_tensor_dist_attr_for_program(
+                            self._dist_context.set_tensor_dist_attr_for_program(
-                            input_var, input_var_attr
+                                input_var, input_var_attr
-                        )
+                            )
                    self._dist_context.set_op_dist_attr_for_program(
                        op, op_dist_attr