Skip to content

M
mindspore

magicwindyyd / mindspore
与 Fork 源项目一致

Fork自 MindSpore / mindspore

通知 1
Star 1
Fork 0
M
mindspore
提交 0ef79831 编写于 作者: M mindspore-ci-bot 提交者： Gitee
浏览文件
操作

!520 Add aicpu support mstype 

Merge pull request !520 from zjun/add_aicpu_support_mstype
上级 40e7b145 ab7e0058
隐藏空白更改
内联 并排
Showing with 3 addition and 0 deletion
mindspore/ccsrc/kernel/aicpu/aicpu_util.cc
浏览文件 @ 0ef79831
......@@ -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.
先完成此消息的编辑！
想要评论请 注册 或