fix paddle2cinn float16 type support bug (PaddlePaddle#48249)

lxsbupt · Dec 8, 2022 · 1569f19 · 1569f19
1 parent 4385ad1
commit 1569f19
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/paddle/fluid/framework/paddle2cinn/transform_type.cc b/paddle/fluid/framework/paddle2cinn/transform_type.cc
@@ -60,6 +60,9 @@ ::phi::DataType TransToPaddleDataType(const cinn_type_t& type) {
   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;