!6945 enable reduce precision support int64 to int32

Merge pull request !6945 from liubuyu/master
2020-09-28 09:29:37 +08:00 · 2020-09-28 09:29:37 +08:00 · badb6748e3
parent d15febdb91 b0837d1137
commit badb6748e3
1 changed files with 6 additions and 0 deletions
--- 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;
  }