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

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

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