forked from lcompilers/lpython
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_tests.py
executable file
·121 lines (104 loc) · 3.88 KB
/
run_tests.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
#!/usr/bin/env python
import sys
import os
ROOT_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__)))
sys.path.append(os.path.join(ROOT_DIR, "src", "libasr"))
from compiler_tester.tester import color, fg, log, run_test, style, tester_main
def single_test(test, verbose, no_llvm, update_reference,
specific_backends=None, excluded_backends=None):
filename = test["filename"]
def is_included(backend):
return test.get(backend, False) \
and (specific_backends is None or backend in specific_backends) \
and (excluded_backends is None or backend not in excluded_backends)
show_verbose = "" if not verbose else "-v"
tokens = is_included("tokens")
ast = is_included("ast")
ast_new = is_included("ast_new")
asr = is_included("asr")
llvm = is_included("llvm")
llvm_dbg = is_included("llvm_dbg")
cpp = is_included("cpp")
c = is_included("c")
wat = is_included("wat")
run = is_included("run")
pass_ = test.get("pass", None)
optimization_passes = ["flip_sign", "div_to_mul", "fma", "sign_from_value",
"inline_function_calls", "loop_unroll",
"dead_code_removal", "loop_vectorise", "print_list"]
if pass_ and (pass_ not in ["do_loops", "global_stmts"] and
pass_ not in optimization_passes):
raise Exception(f"Unknown pass: {pass_}")
log.debug(f"{color(style.bold)} START TEST: {color(style.reset)} {filename}")
extra_args = f"--no-error-banner {show_verbose}"
if tokens:
run_test(
filename,
"tokens",
"lpython --no-color --show-tokens {infile} -o {outfile}",
filename,
update_reference,
extra_args)
if ast:
run_test(
filename,
"ast",
"lpython --show-ast --no-color {infile} -o {outfile}",
filename,
update_reference,
extra_args)
if ast_new:
run_test(
filename,
"ast_new",
"lpython --show-ast --new-parser --no-color {infile} -o {outfile}",
filename,
update_reference,
extra_args)
if asr:
run_test(
filename,
"asr",
"lpython --show-asr --no-color {infile} -o {outfile}",
filename,
update_reference,
extra_args)
if pass_ is not None:
cmd = "lpython --pass=" + pass_ + \
" --show-asr --no-color {infile} -o {outfile}"
run_test(filename, "pass_{}".format(pass_), cmd,
filename, update_reference, extra_args)
if no_llvm:
log.info(f"{filename} * llvm SKIPPED as requested")
else:
if llvm:
run_test(
filename,
"llvm",
"lpython --no-color --show-llvm {infile} -o {outfile}",
filename,
update_reference,
extra_args)
if llvm_dbg:
run_test(
filename,
"llvm_dbg",
"lpython --no-color --show-llvm -g --debug-with-line-column "
"{infile} -o {outfile}",
filename,
update_reference,
extra_args)
if cpp:
run_test(filename, "cpp", "lpython --no-color --show-cpp {infile}",
filename, update_reference, extra_args)
if c:
run_test(filename, "c", "lpython --no-color --show-c {infile}",
filename, update_reference, extra_args)
if wat:
run_test(filename, "wat", "lpython --no-color --show-wat {infile}",
filename, update_reference, extra_args)
if run:
run_test(filename, "runtime", "lpython {infile}",
filename, update_reference, extra_args)
if __name__ == "__main__":
tester_main("LPython", single_test)