From a9e09241ca0d2cd33befe20e301be0486fcce05a Mon Sep 17 00:00:00 2001 From: Simon Pilgrim Date: Thu, 5 Apr 2018 09:50:58 +0000 Subject: [UpdateTestChecks] Make add_asm_checks more like add_ir_checks Towards merging them as mentioned on D45272 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@329265 91177308-0d34-0410-b5e6-96231b3b80d8 --- utils/UpdateTestChecks/asm.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) (limited to 'utils/UpdateTestChecks') diff --git a/utils/UpdateTestChecks/asm.py b/utils/UpdateTestChecks/asm.py index 9d88a266162..aa7e245eb8c 100644 --- a/utils/UpdateTestChecks/asm.py +++ b/utils/UpdateTestChecks/asm.py @@ -215,9 +215,12 @@ def build_function_body_dictionary_for_triple(args, raw_tool_output, triple, pre ##### Generator of assembly CHECK lines -def add_asm_checks(output_lines, comment_marker, run_list, func_dict, func_name): +def add_asm_checks(output_lines, comment_marker, prefix_list, func_dict, func_name): + # Label format is based on ASM string. + check_label_format = '{} %s-LABEL: %s:'.format(comment_marker) + printed_prefixes = [] - for p in run_list: + for p in prefix_list: checkprefixes = p[0] for checkprefix in checkprefixes: if checkprefix in printed_prefixes: @@ -230,11 +233,11 @@ def add_asm_checks(output_lines, comment_marker, run_list, func_dict, func_name) if len(printed_prefixes) != 0: output_lines.append(comment_marker) printed_prefixes.append(checkprefix) - output_lines.append('%s %s-LABEL: %s:' % (comment_marker, checkprefix, func_name)) + output_lines.append(check_label_format % (checkprefix, func_name)) func_body = func_dict[checkprefix][func_name].splitlines() + output_lines.append('%s %s: %s' % (comment_marker, checkprefix, func_body[0])) for func_line in func_body[1:]: output_lines.append('%s %s-NEXT: %s' % (comment_marker, checkprefix, func_line)) - # Add space between different check prefixes and the first line of code. - # output_lines.append(';') + break -- cgit v1.2.3