未验证 提交 73bff10f 编写于 作者: J jiangcheng 提交者: GitHub

fix paddle2cinn float16 type support bug (#48249)

上级 55428765
...@@ -60,6 +60,9 @@ namespace paddle::framework::paddle2cinn { ...@@ -60,6 +60,9 @@ namespace paddle::framework::paddle2cinn {
SET_TYPE_CASE_ITEM(cinn_uint64_t, UINT64) SET_TYPE_CASE_ITEM(cinn_uint64_t, UINT64)
SET_TYPE_CASE_ITEM(cinn_float32_t, FLOAT32) SET_TYPE_CASE_ITEM(cinn_float32_t, FLOAT32)
SET_TYPE_CASE_ITEM(cinn_float64_t, FLOAT64) 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")); PADDLE_THROW(platform::errors::Unimplemented("Input type not supported yet"));
return ::phi::DataType::UNDEFINED; return ::phi::DataType::UNDEFINED;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册