Browse Source

add output score when matched output format with infer

tags/v1.0.0
WilliamLian 5 years ago
parent
commit
3b1dded345
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc

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

@@ -147,6 +147,9 @@ void UpdateCurMatchCounts(const kernel::KernelBuildInfo &kernel_build_info, cons
if (kernel_build_info.GetInputFormat(input_index) == pri_match_format) {
(*cur_kernelinfo_match_counts)[MATCH_SPECIAL_FORMAT_COUNT] += base_score;
}
if (kernel_build_info.GetOutputFormat(input_index) == pri_match_format) {
(*cur_kernelinfo_match_counts)[MATCH_SPECIAL_FORMAT_COUNT] += base_score;
}
if (kernel_build_info.GetInputFormat(input_index) == kOpFormat_DEFAULT) {
(*cur_kernelinfo_match_counts)[MATCH_DEFAULT_FORMAT_COUNT] += base_score;
}


Loading…
Cancel
Save