fix bug of parallel tuner

PaddlePaddle · May 25, 2022 · 1c84d7a · 1c84d7a
1 parent 2f93dac
commit 1c84d7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/paddle/distributed/auto_parallel/tuner/parallel_tuner.py b/python/paddle/distributed/auto_parallel/tuner/parallel_tuner.py
@@ -560,7 +560,7 @@ def _amend_dist_attr(self):
                 for i, dim_mapping in enumerate(dims_mapping):
                     if dim_mapping != -1 \
                         and tensor_shape[i] % process_shape[dim_mapping] != 0:
-                        dims_mapping[i] = [-1]
+                        dims_mapping[i] = -1
 
             for arg_name in dist_attr.outputs_dist_attrs.keys():
                 new_dims_mapping = []
@@ -585,7 +585,7 @@ def _amend_dist_attr(self):
                 for i, dim_mapping in enumerate(dims_mapping):
                     if dim_mapping != -1 \
                         and tensor_shape[i] % process_shape[dim_mapping] != 0:
-                        dims_mapping[i] = [-1]
+                        dims_mapping[i] = -1
 
             dist_op_impls = find_compatible_distributed_operator_impls(
                 dist_op, partial=False)