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
|
#!/usr/bin/env python
import glob
import collections
import os
armcv_path = "arm_compute"
src_path ="src"
Target = collections.namedtuple('Target', 'name prefix basepath')
core_targets = [
Target("NEON", "NE", src_path), # Neon kernels are under src
Target("CL", "CL", src_path), # CL kernels are under src
Target("CPP", "CPP", armcv_path), # CPP kernels are under arm_compute
Target("GLES_COMPUTE", "GC", armcv_path) # GLES kernels are under arm_compute
]
# All functions are under arm_compute
runtime_targets = [
Target("NEON", "NE", armcv_path),
Target("CL", "CL", armcv_path),
Target("CPP", "CPP", armcv_path),
Target("GLES_COMPUTE", "GC", armcv_path)
]
core_path = "/core/"
runtime_path = "/runtime/"
include_str = "#include \""
def read_file(file):
with open(file, "r") as f:
lines = f.readlines()
return lines
def write_file(file, lines):
with open(file, "w") as f:
for line in lines:
f.write(line)
def remove_existing_includes(lines):
first_pos = next(i for i, line in enumerate(lines) if include_str in line)
return [x for x in lines if not x.startswith(include_str)], first_pos
def add_updated_includes(lines, pos, includes):
lines[pos:pos] = includes
return lines
def create_include_list(folder):
files_path = folder + "/*.h"
files = glob.glob(files_path)
updated_files = [include_str + folder + "/" + x.rsplit('/',1)[1] + "\"\n" for x in files]
updated_files.sort()
return updated_files
def include_components(target, path, header_prefix, folder, subfolders=None):
for t in target:
target_path = t.basepath + path + t.name + "/"
components_file = target_path + t.prefix + header_prefix
if os.path.exists(components_file):
include_list = create_include_list(target_path + folder)
for s in subfolders or []:
include_list += create_include_list( target_path + folder + "/" + s)
include_list.sort()
lines = read_file(components_file)
lines, first_pos = remove_existing_includes(lines)
lines = add_updated_includes(lines, first_pos, include_list)
write_file(components_file, lines)
if __name__ == "__main__":
# Include kernels
include_components(core_targets, core_path, "Kernels.h", "kernels", ["arm32", "arm64"])
# Include functions
include_components(runtime_targets, runtime_path, "Functions.h", "functions")
|
