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
|
#!/usr/bin/env python3
###############################################################################
# Copyright (c) Intel Corporation - All rights reserved. #
# This file is part of the LIBXSMM library. #
# #
# For information on the license, see the LICENSE file. #
# Further information: https://github.com/hfp/libxsmm/ #
# SPDX-License-Identifier: BSD-3-Clause #
###############################################################################
# Hans Pabst (Intel Corp.)
###############################################################################
from string import Template
import datetime
import libxsmm_utilities
import fnmatch
import os
import time
import sys
if __name__ == "__main__":
argc = len(sys.argv)
if 1 < argc:
# required argument(s)
filename = sys.argv[1]
# default configuration if no arguments are given
ilp64 = offload = precision = flags = threshold = 0
sync = jit = 1
alpha = beta = 1
cacheline = 64
prefetch = -1
wrap = 1
malloc = 0
mnklist = list()
# optional argument(s)
if 2 < argc:
ilp64 = int(sys.argv[2])
if 3 < argc:
offload = int(sys.argv[3])
if 4 < argc:
cacheline = libxsmm_utilities.sanitize_alignment(int(sys.argv[4]))
if 5 < argc:
precision = int(sys.argv[5])
if 6 < argc:
prefetch = int(sys.argv[6])
if 7 < argc:
threshold = int(sys.argv[7])
if 8 < argc:
sync = int(sys.argv[8])
if 9 < argc:
jit = int(sys.argv[9])
if 10 < argc:
flags = int(sys.argv[10])
if 11 < argc:
alpha = int(sys.argv[11])
if 12 < argc:
beta = int(sys.argv[12])
if 13 < argc:
wrap = int(sys.argv[13])
if 14 < argc:
malloc = int(sys.argv[14])
if 15 < argc:
mnklist = sorted(libxsmm_utilities.load_mnklist(sys.argv[15:], 0))
version, branch, realversion = libxsmm_utilities.version_branch()
major, minor, update, patch = libxsmm_utilities.version_numbers(
version
)
if 0 == threshold:
threshold = 64 * 64 * 64
maxmnk = libxsmm_utilities.max_mnk(mnklist, threshold)
maxdim = int(maxmnk ** (1.0 / 3.0) + 0.5)
avgdim = int(0.5 * maxdim + 0.5)
avgm = libxsmm_utilities.median(
list(map(lambda mnk: mnk[0], mnklist)), avgdim, False
)
avgn = libxsmm_utilities.median(
list(map(lambda mnk: mnk[1], mnklist)), avgdim, False
)
avgk = libxsmm_utilities.median(
list(map(lambda mnk: mnk[2], mnklist)), avgdim, False
)
maxm = libxsmm_utilities.max_mnk(mnklist, avgdim, 0)
maxn = libxsmm_utilities.max_mnk(mnklist, avgdim, 1)
maxk = libxsmm_utilities.max_mnk(mnklist, avgdim, 2)
build_date = datetime.datetime.utcfromtimestamp(
int(os.environ.get('SOURCE_DATE_EPOCH', time.time()))
)
substitute = {
"VERSION": realversion,
"BRANCH": branch,
"MAJOR": major,
"MINOR": minor,
"UPDATE": update,
"PATCH": patch,
"DATE": build_date.strftime("%Y%m%d"),
"CACHELINE": cacheline,
"PREFETCH": [-1, prefetch][0 <= prefetch],
"MAX_MNK": maxmnk,
"MAX_DIM": maxdim,
"AVG_DIM": int((maxdim + 1) / 2),
"MAX_M": [maxdim, maxm][avgm < maxm],
"MAX_N": [maxdim, maxn][avgn < maxn],
"MAX_K": [maxdim, maxk][avgk < maxk],
"FLAGS": flags,
"ILP64": [0, 1][0 != ilp64],
"ALPHA": alpha,
"BETA": beta,
"WRAP": wrap,
"MALLOC": malloc,
"SYNC": [0, 1][0 != sync],
"JIT": [0, 1][0 != jit],
"LIBXSMM_OFFLOAD_BUILD": ["", "\n#define LIBXSMM_OFFLOAD_BUILD"][
0 != offload
],
"MNK_PREPROCESSOR_LIST": "",
}
template = Template(open(filename, "r").read())
if fnmatch.fnmatch(filename, "*.h*"):
if mnklist:
first = mnklist[0]
for mnk in mnklist:
mnkstr = "_".join(map(str, mnk))
if mnk != first:
substitute["MNK_PREPROCESSOR_LIST"] += "\n"
if 2 != precision:
substitute["MNK_PREPROCESSOR_LIST"] += (
"#define LIBXSMM_SMM_" + mnkstr
)
if mnk != first or 0 == precision:
substitute["MNK_PREPROCESSOR_LIST"] += "\n"
if 1 != precision:
substitute["MNK_PREPROCESSOR_LIST"] += (
"#define LIBXSMM_DMM_" + mnkstr
)
print(template.substitute(substitute))
else:
substitute["BLASINT_KIND"] = ["C_INT", "C_LONG_LONG"][0 != ilp64]
print(template.safe_substitute(substitute))
else:
sys.tracebacklimit = 0
raise ValueError(sys.argv[0] + ": wrong number of arguments!")
|
