diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc index 39cec09e3db86..0f2833db8091f 100644 --- a/paddle/phi/infermeta/unary.cc +++ b/paddle/phi/infermeta/unary.cc @@ -358,6 +358,7 @@ void AsComplexInferMeta(const MetaTensor& input, MetaTensor* output) { const phi::DDim out_dims(in_dims.Get(), input_rank - 1); output->set_dims(out_dims); output->share_lod(input); + output->set_dtype(dtype::ToComplex(input.dtype())); } void BatchSizeLikeInferMeta(const MetaTensor& x, diff --git a/test/legacy_test/test_complex_view_op.py b/test/legacy_test/test_complex_view_op.py index c529e3950a9fb..75ebe52bd7ec4 100644 --- a/test/legacy_test/test_complex_view_op.py +++ b/test/legacy_test/test_complex_view_op.py @@ -50,6 +50,7 @@ def test_check_grad(self): self.check_grad( ['X'], 'Out', + check_pir=True, ) @@ -71,6 +72,7 @@ def test_check_grad(self): self.check_grad( ['X'], 'Out', + check_pir=True, )