diff --git a/mindspore/ccsrc/kernel/aicpu/aicpu_util.cc b/mindspore/ccsrc/kernel/aicpu/aicpu_util.cc
index 316df639226f2ab1752f899ff363fca69d2f0ad2..a617f56f8f0fcefbc293fa2b3bfd7453d721c9aa 100644
--- a/mindspore/ccsrc/kernel/aicpu/aicpu_util.cc
+++ b/mindspore/ccsrc/kernel/aicpu/aicpu_util.cc
@@ -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},
 };