diff --git a/paddle/fluid/framework/paddle2cinn/transform_type.cc b/paddle/fluid/framework/paddle2cinn/transform_type.cc index 60502edd99acf94de2d0cdada3c3545692ec5f81..c8172bd3f55f1638ee69d62828eaec0f81617759 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;