|
|
|
@@ -147,9 +147,6 @@ 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; |
|
|
|
} |
|
|
|
@@ -161,6 +158,9 @@ void UpdateCurMatchCounts(const kernel::KernelBuildInfo &kernel_build_info, cons |
|
|
|
AnfAlgo::GetOutputInferDataType(kernel_node, output_index)) { |
|
|
|
(*cur_kernelinfo_match_counts)[MATCH_OUTPUT_DTYPE_COUNT] += 1; |
|
|
|
} |
|
|
|
if (kernel_build_info.GetOutputFormat(output_index) == pri_match_format) { |
|
|
|
(*cur_kernelinfo_match_counts)[MATCH_SPECIAL_FORMAT_COUNT] += 1; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|