Browse Source

!520 Add aicpu support mstype

Merge pull request !520 from zjun/add_aicpu_support_mstype
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
0ef7983174
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/kernel/aicpu/aicpu_util.cc

+ 3
- 0
mindspore/ccsrc/kernel/aicpu/aicpu_util.cc View File

@@ -27,6 +27,7 @@ namespace kernel {
static std::map<int32_t, int32_t> MS_PROTO_DATA_TYPE_MAP = { static std::map<int32_t, int32_t> MS_PROTO_DATA_TYPE_MAP = {
{mindspore::TypeId::kTypeUnknown, mindspore::DataType::MS_UNKNOWN}, {mindspore::TypeId::kTypeUnknown, mindspore::DataType::MS_UNKNOWN},
{mindspore::TypeId::kNumberTypeBool, mindspore::DataType::MS_BOOL}, {mindspore::TypeId::kNumberTypeBool, mindspore::DataType::MS_BOOL},
{mindspore::TypeId::kNumberTypeInt, mindspore::DataType::MS_INT32},
{mindspore::TypeId::kNumberTypeInt8, mindspore::DataType::MS_INT8}, {mindspore::TypeId::kNumberTypeInt8, mindspore::DataType::MS_INT8},
{mindspore::TypeId::kNumberTypeInt16, mindspore::DataType::MS_INT16}, {mindspore::TypeId::kNumberTypeInt16, mindspore::DataType::MS_INT16},
{mindspore::TypeId::kNumberTypeInt32, mindspore::DataType::MS_INT32}, {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::kNumberTypeUInt, mindspore::DataType::MS_UINT32},
{mindspore::TypeId::kNumberTypeUInt8, mindspore::DataType::MS_UINT8}, {mindspore::TypeId::kNumberTypeUInt8, mindspore::DataType::MS_UINT8},
{mindspore::TypeId::kNumberTypeUInt16, mindspore::DataType::MS_UINT16}, {mindspore::TypeId::kNumberTypeUInt16, mindspore::DataType::MS_UINT16},
{mindspore::TypeId::kNumberTypeUInt32, mindspore::DataType::MS_UINT32},
{mindspore::TypeId::kNumberTypeUInt64, mindspore::DataType::MS_UINT64}, {mindspore::TypeId::kNumberTypeUInt64, mindspore::DataType::MS_UINT64},
{mindspore::TypeId::kNumberTypeFloat16, mindspore::DataType::MS_FLOAT16}, {mindspore::TypeId::kNumberTypeFloat16, mindspore::DataType::MS_FLOAT16},
{mindspore::TypeId::kNumberTypeFloat, mindspore::DataType::MS_FLOAT32},
{mindspore::TypeId::kNumberTypeFloat32, mindspore::DataType::MS_FLOAT32}, {mindspore::TypeId::kNumberTypeFloat32, mindspore::DataType::MS_FLOAT32},
{mindspore::TypeId::kNumberTypeFloat64, mindspore::DataType::MS_FLOAT64}, {mindspore::TypeId::kNumberTypeFloat64, mindspore::DataType::MS_FLOAT64},
}; };


Loading…
Cancel
Save