【prim】test composite rules with -1 shape (#51435)

* init * modify
PaddlePaddle · Mar 14, 2023 · 82a7c33 · 82a7c33
1 parent c3f8ba9
commit 82a7c33
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/paddle/incubate/autograd/primx.py b/python/paddle/incubate/autograd/primx.py
@@ -634,6 +634,9 @@ def expand_nested_list(xs):
                             f'but orig_out.dtype={orig_out.dtype} and new_out.dtype={new_out.dtype}'
                         )
                         if orig_out.shape and new_out.shape:
+                            assert (
+                                -1 not in new_out.shape
+                            ), f'when replace origin op {op_name} with composite rule, composite out shape has -1.'
                             assert orig_out.shape == new_out.shape, (
                                 f'when replace origin op {op_name} with composite rule, origin out shape should be equal to new out shape, '
                                 f'but orig_out.shape={orig_out.shape} and new_out.shape={new_out.shape}'