[AutoParallel] fix process_mesh's member (#49371)

836a662c · zhaoyingli · GitHub · 02019804 · 836a662c
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

python/paddle/distributed/passes/auto_parallel_quantization.py ...n/paddle/distributed/passes/auto_parallel_quantization.py +2 -2

未找到文件。
--- 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_shape": var_dist_attr.process_mesh.shape,
-                "process_group": var_dist_attr.process_mesh.processes,
+                "process_group": var_dist_attr.process_mesh.process_ids,
            }
            # slice tensor_value with dist_attr