Allow update_test_checks.py to not scrub names.

Add a --preserve-names option to tell the script not to replace IR names.
Sometimes tests want those names.  For example if a test is looking for a
modification to an existing instruction we'll want to make the names.

Differential Revision: https://reviews.llvm.org/D68081

llvm-svn: 373912
This commit is contained in:
David Greene 2019-10-07 14:37:20 +00:00
parent dc4d908d6e
commit a14ffc7eb7
2 changed files with 8 additions and 3 deletions

View File

@ -267,10 +267,12 @@ def add_checks(output_lines, comment_marker, prefix_list, func_dict, func_name,
output_lines.append(comment_marker)
break
def add_ir_checks(output_lines, comment_marker, prefix_list, func_dict, func_name):
def add_ir_checks(output_lines, comment_marker, prefix_list, func_dict,
func_name, preserve_names):
# Label format is based on IR string.
check_label_format = '{} %s-LABEL: @%s('.format(comment_marker)
add_checks(output_lines, comment_marker, prefix_list, func_dict, func_name, check_label_format, False, False)
add_checks(output_lines, comment_marker, prefix_list, func_dict, func_name,
check_label_format, False, preserve_names)
def add_analyze_checks(output_lines, comment_marker, prefix_list, func_dict, func_name):
check_label_format = '{} %s-LABEL: \'%s\''.format(comment_marker)

View File

@ -64,6 +64,8 @@ def main():
'--function', help='The function in the test file to update')
parser.add_argument('-u', '--update-only', action='store_true',
help='Only update test if it was already autogened')
parser.add_argument('-p', '--preserve-names', action='store_true',
help='Do not scrub IR names')
parser.add_argument('tests', nargs='+')
args = parser.parse_args()
@ -174,7 +176,8 @@ def main():
continue
# Print out the various check lines here.
common.add_ir_checks(output_lines, ';', prefix_list, func_dict, func_name)
common.add_ir_checks(output_lines, ';', prefix_list, func_dict,
func_name, args.preserve_names)
is_in_function_start = False
if is_in_function: