From 3b1dded345c834bdce44b9b2b01317dd388e8ca5 Mon Sep 17 00:00:00 2001 From: WilliamLian Date: Thu, 10 Sep 2020 09:56:49 +0800 Subject: [PATCH] add output score when matched output format with infer --- mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc b/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc index 9189741add..b7fc48f1f0 100644 --- a/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc +++ b/mindspore/ccsrc/runtime/device/ascend/kernel_select_ascend.cc @@ -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; }