diff --git a/llvm/test/tools/UpdateTestChecks/update_mir_test_checks/Inputs/x86-multiple-prefixes.ll.expected b/llvm/test/tools/UpdateTestChecks/update_mir_test_checks/Inputs/x86-multiple-prefixes.ll.expected --- a/llvm/test/tools/UpdateTestChecks/update_mir_test_checks/Inputs/x86-multiple-prefixes.ll.expected +++ b/llvm/test/tools/UpdateTestChecks/update_mir_test_checks/Inputs/x86-multiple-prefixes.ll.expected @@ -22,6 +22,7 @@ ; NOAVX-NEXT: [[MOV64rm:%[0-9]+]]:gr64 = MOV64rm $rip, 1, $noreg, target-flags(x86-gotpcrel) @z, $noreg :: (load (s64) from got, align 16) ; NOAVX-NEXT: MOVAPSmr killed [[MOV64rm]], 1, $noreg, 0, $noreg, killed [[V_SET0_]] :: (store (s128) into @z) ; NOAVX-NEXT: RET 0 + ; ; AVX-LABEL: name: zero128 ; AVX: bb.0 (%ir-block.0): ; AVX-NEXT: [[V_SET0_:%[0-9]+]]:vr128 = V_SET0 diff --git a/llvm/utils/update_mir_test_checks.py b/llvm/utils/update_mir_test_checks.py --- a/llvm/utils/update_mir_test_checks.py +++ b/llvm/utils/update_mir_test_checks.py @@ -235,9 +235,10 @@ break if not func_dict[prefix][func_name]: continue - # if printed_prefixes: - # # Add some space between different check prefixes. - # output_lines.append('') + if printed_prefixes: + # Add some space between different check prefixes. + indent = len(output_lines[-1]) - len(output_lines[-1].lstrip(" ")) + output_lines.append(" "*indent + ";") printed_prefixes.add(prefix) log("Adding {} lines for {}".format(prefix, func_name), args.verbose) add_check_lines( @@ -333,9 +334,9 @@ def should_add_line_to_output(input_line, prefix_set): - # Skip any check lines that we're handling. + # Skip any check lines that we're handling as well as comments m = common.CHECK_RE.match(input_line) - if m and m.group(1) in prefix_set: + if (m and m.group(1) in prefix_set) or re.search("^[ \t]*;", input_line): return False return True