Skip to content

Commit 73bff10

Browse files
authored
fix paddle2cinn float16 type support bug (#48249)
1 parent 5542876 commit 73bff10

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

paddle/fluid/framework/paddle2cinn/transform_type.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,9 @@ ::phi::DataType TransToPaddleDataType(const cinn_type_t& type) {
6060
SET_TYPE_CASE_ITEM(cinn_uint64_t, UINT64)
6161
SET_TYPE_CASE_ITEM(cinn_float32_t, FLOAT32)
6262
SET_TYPE_CASE_ITEM(cinn_float64_t, FLOAT64)
63+
#ifdef CINN_COMMON_FLOAT16_H
64+
SET_TYPE_CASE_ITEM(cinn_float16_t, FLOAT16)
65+
#endif // CINN_COMMON_FLOAT16_H
6366

6467
PADDLE_THROW(platform::errors::Unimplemented("Input type not supported yet"));
6568
return ::phi::DataType::UNDEFINED;

0 commit comments

Comments
 (0)