From 610d2e424e57faf92da55768e8050966842253f7 Mon Sep 17 00:00:00 2001 From: Simon Pilgrim Date: Sat, 2 Mar 2019 11:14:01 +0000 Subject: [PATCH] Fix update_mir_test_checks.py to run on python3 Split off from D58817 Differential Revision: https://reviews.llvm.org/D58820 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@355268 91177308-0d34-0410-b5e6-96231b3b80d8 --- utils/update_mir_test_checks.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/utils/update_mir_test_checks.py b/utils/update_mir_test_checks.py index 56d236de249..cf7528f5760 100755 --- a/utils/update_mir_test_checks.py +++ b/utils/update_mir_test_checks.py @@ -62,6 +62,8 @@ class LLC: with open(ir) as ir_file: stdout = subprocess.check_output('{} {}'.format(self.bin, args), shell=True, stdin=ir_file) + if sys.version_info[0] > 2: + stdout = stdout.decode() # Fix line endings to unix CR style. stdout = stdout.replace('\r\n', '\n') return stdout @@ -408,7 +410,7 @@ def update_test_file(llc, test, remove_common_prefixes=False, verbose=False): log('Writing {} lines to {}...'.format(len(output_lines), test), verbose) with open(test, 'wb') as fd: - fd.writelines([l + '\n' for l in output_lines]) + fd.writelines(['{}\n'.format(l).encode('utf-8') for l in output_lines]) def main(): -- 2.50.1