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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
|
#!/usr/bin/env python3
"""This script verifies expression folding.
It compiles a source file with '-fdebug-dump-symbols'
and looks for parameter declarations to check
they have been folded as expected.
To check folding of an expression EXPR,
the fortran program passed to this script
must contain the following:
logical, parameter :: test_x = <compare EXPR to expected value>
This script will test that all parameter
with a name starting with "test_"
have been folded to .true.
For instance, acos folding can be tested with:
real(4), parameter :: res_acos = acos(0.5_4)
real(4), parameter :: exp_acos = 1.047
logical, parameter :: test_acos = abs(res_acos - exp_acos).LE.(0.001_4)
There are two kinds of failure:
- test_x is folded to .false..
This means the expression was folded
but the value is not as expected.
- test_x is not folded (it is neither .true. nor .false.).
This means the compiler could not fold the expression.
Parameters:
sys.argv[1]: a source file with contains the input and expected output
sys.argv[2]: the Flang frontend driver
sys.argv[3:]: Optional arguments to the Flang frontend driver"""
import os
import sys
import tempfile
import re
import subprocess
from difflib import unified_diff
from pathlib import Path
def check_args(args):
"""Verifies that the number is arguments passed is correct."""
if len(args) < 3:
print(f"Usage: {args[0]} <fortran-source> <flang-command>")
sys.exit(1)
def set_source(source):
"""Sets the path to the source files."""
if not Path(source).is_file():
print(f"File not found: {src}")
sys.exit(1)
return Path(source)
def set_executable(exe):
"""Sets the path to the Flang frontend driver."""
if not Path(exe).is_file():
print(f"Flang was not found: {exe}")
sys.exit(1)
return str(Path(exe))
check_args(sys.argv)
cwd = os.getcwd()
srcdir = set_source(sys.argv[1]).resolve()
with open(srcdir, "r", encoding="utf-8") as f:
src = f.readlines()
src1 = ""
src2 = ""
src3 = ""
src4 = ""
messages = ""
actual_warnings = ""
expected_warnings = ""
warning_diffs = ""
flang_fc1 = set_executable(sys.argv[2])
flang_fc1_args = sys.argv[3:]
flang_fc1_options = ""
LIBPGMATH = os.getenv("LIBPGMATH")
if LIBPGMATH:
flang_fc1_options = ["-fdebug-dump-symbols", "-DTEST_LIBPGMATH"]
print("Assuming libpgmath support")
else:
flang_fc1_options = ["-fdebug-dump-symbols"]
print("Not assuming libpgmath support")
cmd = [flang_fc1, *flang_fc1_args, *flang_fc1_options, str(srcdir)]
with tempfile.TemporaryDirectory() as tmpdir:
proc = subprocess.run(
cmd,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
check=True,
universal_newlines=True,
cwd=tmpdir,
)
src1 = proc.stdout
messages = proc.stderr
for line in src1.split("\n"):
m = re.search(r"(\w*)(?=, PARAMETER).*init:(.*)", line)
if m:
src2 += f"{m.group(1)} {m.group(2)}\n"
for line in src2.split("\n"):
m = re.match(r"test_*", line)
if m:
src3 += f"{m.string}\n"
for passed_results, line in enumerate(src3.split("\n")):
m = re.search(r"\.false\._.$", line)
if m:
src4 += f"{line}\n"
for line in messages.split("\n"):
m = re.search(r"[^:]*:(\d*):\d*: (.*)", line)
if m:
actual_warnings += f"{m.group(1)}: {m.group(2)}\n"
passed_warnings = 0
warnings = []
for i, line in enumerate(src, 1):
m = re.search(r"(?:!WARN:)(.*)", line)
if m:
warnings.append(m.group(1))
continue
if warnings:
for x in warnings:
passed_warnings += 1
expected_warnings += f"{i}:{x}\n"
warnings = []
for line in unified_diff(
actual_warnings.split("\n"), expected_warnings.split("\n"), n=0
):
line = re.sub(r"(^\-)(\d+:)", r"\nactual at \g<2>", line)
line = re.sub(r"(^\+)(\d+:)", r"\nexpect at \g<2>", line)
warning_diffs += line
if src4 or warning_diffs:
print("Folding test failed:")
# Prints failed tests, including parameters with the same
# suffix so that more information can be obtained by declaring
# expected_x and result_x
if src4:
for line in src4.split("\n"):
m = re.match(r"test_(\w+)", line)
if m:
for line in src2.split("\n"):
if m.group(1) in line:
print(line)
if warning_diffs:
print(warning_diffs)
print()
print("FAIL")
sys.exit(1)
else:
print()
print(f"All {passed_results+passed_warnings} tests passed")
print("PASS")
|
