diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py index 3858f38c10b44c..5e7c0382046785 100644 --- 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 index c109e41aeddfa1..99350c9eb043e4 100755 --- 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 []