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
166
|
"""
Generate CPU features tables from CCompilerOpt
"""
from os import sys, path
gen_path = path.dirname(path.realpath(__file__))
from numpy.distutils.ccompiler_opt import CCompilerOpt
class FakeCCompilerOpt(CCompilerOpt):
fake_info = ""
def __init__(self, *args, **kwargs):
no_cc = None
CCompilerOpt.__init__(self, no_cc, **kwargs)
def dist_compile(self, sources, flags, **kwargs):
return sources
def dist_info(self):
return FakeCCompilerOpt.fake_info
@staticmethod
def dist_log(*args, stderr=False):
# avoid printing
pass
def feature_test(self, name, force_flags=None):
# To speed up
return True
def gen_features_table(self, features, ignore_groups=True,
field_names=["Name", "Implies"], **kwargs):
rows = []
for f in features:
is_group = "group" in self.feature_supported.get(f, {})
if ignore_groups and is_group:
continue
implies = self.feature_sorted(self.feature_implies(f))
implies = ' '.join(['``%s``' % i for i in implies])
rows.append([f, implies])
return self.gen_rst_table(field_names, rows, **kwargs)
def gen_gfeatures_table(self, features,
field_names=["Name", "Gather", "Implies"],
**kwargs):
rows = []
for f in features:
gather = self.feature_supported.get(f, {}).get("group", None)
if not gather:
continue
implies = self.feature_sorted(self.feature_implies(f))
implies = ' '.join(['``%s``' % i for i in implies])
gather = ' '.join(['``%s``' % i for i in gather])
rows.append([f, gather, implies])
return self.gen_rst_table(field_names, rows, **kwargs)
def gen_rst_table(self, field_names, rows, margin_left=2):
assert(not rows or len(field_names) == len(rows[0]))
rows.append(field_names)
fld_len = len(field_names)
cls_len = [max(len(c[i]) for c in rows) for i in range(fld_len)]
del rows[-1]
padding = 0
cformat = ' '.join('{:<%d}' % (i+padding) for i in cls_len)
border = cformat.format(*['='*i for i in cls_len])
rows = [cformat.format(*row) for row in rows]
# header
rows = [border, cformat.format(*field_names), border] + rows
# footer
rows += [border]
# add left margin
rows = [(' ' * margin_left) + r for r in rows]
return '\n'.join(rows)
if __name__ == '__main__':
margin_left = 4*1
############### x86 ###############
FakeCCompilerOpt.fake_info = "x86_64 gcc"
x64_gcc = FakeCCompilerOpt(cpu_baseline="max")
x86_tables = """\
``X86`` - CPU feature names
~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. table::
:align: left
{x86_features}
``X86`` - Group names
~~~~~~~~~~~~~~~~~~~~~
.. table::
:align: left
{x86_gfeatures}
""".format(
x86_features = x64_gcc.gen_features_table(
x64_gcc.cpu_baseline_names(), margin_left=margin_left
),
x86_gfeatures = x64_gcc.gen_gfeatures_table(
x64_gcc.cpu_baseline_names(), margin_left=margin_left
)
)
############### Power ###############
FakeCCompilerOpt.fake_info = "ppc64 gcc"
ppc64_gcc = FakeCCompilerOpt(cpu_baseline="max")
FakeCCompilerOpt.fake_info = "ppc64le gcc"
ppc64le_gcc = FakeCCompilerOpt(cpu_baseline="max")
ppc64_tables = """\
``IBM/POWER`` ``big-endian`` - CPU feature names
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. table::
:align: left
{ppc64_features}
``IBM/POWER`` ``little-endian mode`` - CPU feature names
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. table::
:align: left
{ppc64le_features}
""".format(
ppc64_features = ppc64_gcc.gen_features_table(
ppc64_gcc.cpu_baseline_names(), margin_left=margin_left
),
ppc64le_features = ppc64le_gcc.gen_features_table(
ppc64le_gcc.cpu_baseline_names(), margin_left=margin_left
)
)
############### Arm ###############
FakeCCompilerOpt.fake_info = "armhf gcc"
armhf_gcc = FakeCCompilerOpt(cpu_baseline="max")
FakeCCompilerOpt.fake_info = "aarch64 gcc"
aarch64_gcc = FakeCCompilerOpt(cpu_baseline="max")
arm_tables = """\
``ARMHF`` - CPU feature names
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. table::
:align: left
{armhf_features}
``ARM64`` ``AARCH64`` - CPU feature names
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. table::
:align: left
{aarch64_features}
""".format(
armhf_features = armhf_gcc.gen_features_table(
armhf_gcc.cpu_baseline_names(), margin_left=margin_left
),
aarch64_features = aarch64_gcc.gen_features_table(
aarch64_gcc.cpu_baseline_names(), margin_left=margin_left
)
)
# TODO: diff the difference among all supported compilers
with open(path.join(gen_path, 'simd-optimizations-tables.inc'), 'wt') as fd:
fd.write(f'.. generated via {__file__}\n\n')
fd.write(x86_tables)
fd.write(ppc64_tables)
fd.write(arm_tables)
|
