[llvm] r329265 - [UpdateTestChecks] Make add_asm_checks more like add_ir_checks

[Simon Pilgrim via llvm-commits]

Author: rksimon
Date: Thu Apr  5 02:50:58 2018
New Revision: 329265

URL: http://llvm.org/viewvc/llvm-project?rev=329265&view=rev
Log:
[UpdateTestChecks] Make add_asm_checks more like add_ir_checks

Towards merging them as mentioned on D45272

Modified:
    llvm/trunk/utils/UpdateTestChecks/asm.py

Modified: llvm/trunk/utils/UpdateTestChecks/asm.py
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/utils/UpdateTestChecks/asm.py?rev=329265&r1=329264&r2=329265&view=diff
==============================================================================
--- llvm/trunk/utils/UpdateTestChecks/asm.py (original)
+++ llvm/trunk/utils/UpdateTestChecks/asm.py Thu Apr  5 02:50:58 2018
@@ -215,9 +215,12 @@ def build_function_body_dictionary_for_t
 
 ##### 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
       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