diff --git a/llvm/utils/update_mir_test_checks.py b/llvm/utils/update_mir_test_checks.py index 56d236de2491..cf7528f57609 100755 --- a/llvm/utils/update_mir_test_checks.py +++ b/llvm/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():