From 836a662cdf5a32659b8ad3438e12be9a2fe77f4c Mon Sep 17 00:00:00 2001
From: zhaoyingli <86812880+zhaoyinglia@users.noreply.github.com>
Date: Wed, 28 Dec 2022 10:55:43 +0800
Subject: [PATCH] [AutoParallel] fix process_mesh's member (#49371)

---
 .../paddle/distributed/passes/auto_parallel_quantization.py   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/paddle/distributed/passes/auto_parallel_quantization.py b/python/paddle/distributed/passes/auto_parallel_quantization.py
index 5ece72458a..924499c273 100644
--- a/python/paddle/distributed/passes/auto_parallel_quantization.py
+++ b/python/paddle/distributed/passes/auto_parallel_quantization.py
@@ -229,8 +229,8 @@ class QuantizationPass(PassBase):
             var_dist_attr = dist_context.get_tensor_dist_attr_for_program(var)
             dist_attr = {
                 "dims_mapping": var_dist_attr.dims_mapping,
-                "process_shape": var_dist_attr.process_mesh.topology,
-                "process_group": var_dist_attr.process_mesh.processes,
+                "process_shape": var_dist_attr.process_mesh.shape,
+                "process_group": var_dist_attr.process_mesh.process_ids,
             }
 
             # slice tensor_value with dist_attr
-- 
GitLab