diff --git a/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc b/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc
index 13817d4e502..5a423a62ae8 100644
--- a/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc
+++ b/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc
@@ -179,6 +179,12 @@ void AddSupportMixedPrecisionDataTypeIndex(TypeId data_type, std::vector<int> *s
     case kNumberTypeFloat:
       index = 1;
       break;
+    case kNumberTypeInt32:
+      index = 0;
+      break;
+    case kNumberTypeInt64:
+      index = 1;
+      break;
     default:
       break;
   }