From a3a2c45bab80712556a1064849d5c5de0f38fab2 Mon Sep 17 00:00:00 2001 From: Chen Weihang Date: Thu, 28 Apr 2022 04:12:29 +0000 Subject: [PATCH] add testcases --- paddle/fluid/framework/infershape_utils.cc | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/paddle/fluid/framework/infershape_utils.cc b/paddle/fluid/framework/infershape_utils.cc index 97366397ee057..089698517a0c9 100644 --- a/paddle/fluid/framework/infershape_utils.cc +++ b/paddle/fluid/framework/infershape_utils.cc @@ -685,6 +685,14 @@ CompatInferMetaContext BuildInferMetaContext(InferShapeContext* ctx, infer_meta_context.EmplaceBackAttr( BOOST_GET_CONST(std::vector, attr)); break; + case phi::AttributeType::BOOLS: + infer_meta_context.EmplaceBackAttr( + BOOST_GET_CONST(std::vector, attr)); + break; + case phi::AttributeType::FLOAT64S: + infer_meta_context.EmplaceBackAttr( + BOOST_GET_CONST(std::vector, attr)); + break; default: PADDLE_THROW(platform::errors::Unimplemented( "Unsupported cast op attribute `%s` when construct "