[Change] reshape dtype in pir mode (PaddlePaddle#58924)

* [Change] reshape dtype in pir mode * [Temp] just test pir works * [Restore] uncomment dtyps
danleifeng · Nov 13, 2023 · 836c161 · 836c161
1 parent 9a5a8f8
commit 836c161
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py
@@ -3857,6 +3857,8 @@ def get_attr_shape(list_shape):
                 'int64',
                 'bool',
                 'uint16',
+                'complex64',
+                'complex128',
             ],
             'reshape',
         )

diff --git a/test/legacy_test/test_reshape_op.py b/test/legacy_test/test_reshape_op.py
@@ -453,6 +453,7 @@ def _test_api(self):
         np.testing.assert_array_equal(res_3, input.reshape([5, 10]))
         np.testing.assert_array_equal(res_4, input.reshape(shape))
 
+    @test_with_pir_api
     def _test_static_dtype(self):
         places = [paddle.CPUPlace()] + (
             [paddle.CUDAPlace(0)] if base.core.is_compiled_with_cuda() else []