From 73bff10ff4a1eff33566e03c499c89f83f8bbe1a Mon Sep 17 00:00:00 2001 From: jiangcheng Date: Thu, 8 Dec 2022 20:27:37 +0800 Subject: [PATCH] fix paddle2cinn float16 type support bug (#48249) --- paddle/fluid/framework/paddle2cinn/transform_type.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/paddle/fluid/framework/paddle2cinn/transform_type.cc b/paddle/fluid/framework/paddle2cinn/transform_type.cc index 60502edd99..c8172bd3f5 100644 --- a/paddle/fluid/framework/paddle2cinn/transform_type.cc +++ b/paddle/fluid/framework/paddle2cinn/transform_type.cc @@ -60,6 +60,9 @@ namespace paddle::framework::paddle2cinn { SET_TYPE_CASE_ITEM(cinn_uint64_t, UINT64) SET_TYPE_CASE_ITEM(cinn_float32_t, FLOAT32) SET_TYPE_CASE_ITEM(cinn_float64_t, FLOAT64) +#ifdef CINN_COMMON_FLOAT16_H + SET_TYPE_CASE_ITEM(cinn_float16_t, FLOAT16) +#endif // CINN_COMMON_FLOAT16_H PADDLE_THROW(platform::errors::Unimplemented("Input type not supported yet")); return ::phi::DataType::UNDEFINED; -- GitLab