未验证 提交 9e73be65 编写于 作者: J jiangcheng 提交者: GitHub

[CINN] support int8/uint8/int16/uint16 dtype (#50566)

上级 b5d0d8c8
......@@ -17,8 +17,8 @@ if(NOT WITH_CINN)
endif()
if(NOT CINN_GIT_TAG)
# 2023.02.09 commit
set(CINN_GIT_TAG 63cf4267a0f8756df61b0dca5fb30bf3748ba2f4)
# 2023.02.16 commit
set(CINN_GIT_TAG ec7f1a87b195b9586e69f9f30a2b647dcf8b9def)
endif()
message(STATUS "CINN version: " ${CINN_GIT_TAG})
......
......@@ -28,14 +28,17 @@ namespace paddle::framework::paddle2cinn {
}
SET_TYPE_CASE_ITEM(Bool, BOOL)
SET_TYPE_CASE_ITEM(I8, INT8)
SET_TYPE_CASE_ITEM(I16, INT16)
SET_TYPE_CASE_ITEM(I32, INT32)
SET_TYPE_CASE_ITEM(I64, INT64)
SET_TYPE_CASE_ITEM(UI8, UINT8)
SET_TYPE_CASE_ITEM(UI16, UINT16)
SET_TYPE_CASE_ITEM(UI32, UINT32)
SET_TYPE_CASE_ITEM(UI64, UINT64)
SET_TYPE_CASE_ITEM(F16, FLOAT16)
SET_TYPE_CASE_ITEM(F32, FLOAT32)
SET_TYPE_CASE_ITEM(F64, FLOAT64)
......@@ -53,11 +56,17 @@ namespace paddle::framework::paddle2cinn {
}
SET_TYPE_CASE_ITEM(cinn_bool_t, BOOL)
SET_TYPE_CASE_ITEM(cinn_int8_t, INT8)
SET_TYPE_CASE_ITEM(cinn_int16_t, INT16)
SET_TYPE_CASE_ITEM(cinn_int32_t, INT32)
SET_TYPE_CASE_ITEM(cinn_int64_t, INT64)
SET_TYPE_CASE_ITEM(cinn_uint8_t, UINT8)
SET_TYPE_CASE_ITEM(cinn_uint16_t, UINT16)
SET_TYPE_CASE_ITEM(cinn_uint32_t, UINT32)
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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册