Browse Source

enable reduce precision support int64 to int32

tags/v1.1.0
liubuyu 5 years ago
parent
commit
b0837d1137
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc

+ 6
- 0
mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc View File

@@ -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;
}


Loading…
Cancel
Save