提交 0ef79831 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!520 Add aicpu support mstype

Merge pull request !520 from zjun/add_aicpu_support_mstype
......@@ -27,6 +27,7 @@ namespace kernel {
static std::map<int32_t, int32_t> MS_PROTO_DATA_TYPE_MAP = {
{mindspore::TypeId::kTypeUnknown, mindspore::DataType::MS_UNKNOWN},
{mindspore::TypeId::kNumberTypeBool, mindspore::DataType::MS_BOOL},
{mindspore::TypeId::kNumberTypeInt, mindspore::DataType::MS_INT32},
{mindspore::TypeId::kNumberTypeInt8, mindspore::DataType::MS_INT8},
{mindspore::TypeId::kNumberTypeInt16, mindspore::DataType::MS_INT16},
{mindspore::TypeId::kNumberTypeInt32, mindspore::DataType::MS_INT32},
......@@ -34,8 +35,10 @@ static std::map<int32_t, int32_t> MS_PROTO_DATA_TYPE_MAP = {
{mindspore::TypeId::kNumberTypeUInt, mindspore::DataType::MS_UINT32},
{mindspore::TypeId::kNumberTypeUInt8, mindspore::DataType::MS_UINT8},
{mindspore::TypeId::kNumberTypeUInt16, mindspore::DataType::MS_UINT16},
{mindspore::TypeId::kNumberTypeUInt32, mindspore::DataType::MS_UINT32},
{mindspore::TypeId::kNumberTypeUInt64, mindspore::DataType::MS_UINT64},
{mindspore::TypeId::kNumberTypeFloat16, mindspore::DataType::MS_FLOAT16},
{mindspore::TypeId::kNumberTypeFloat, mindspore::DataType::MS_FLOAT32},
{mindspore::TypeId::kNumberTypeFloat32, mindspore::DataType::MS_FLOAT32},
{mindspore::TypeId::kNumberTypeFloat64, mindspore::DataType::MS_FLOAT64},
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册