• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1exports_files(["LICENSE"])
2
3cc_library(
4    name = "include",
5    hdrs = glob([
6        "include/**/*.h",
7        "include/**/*.hpp",
8    ]),
9    includes = ["include"],
10    strip_include_prefix = "include",
11)
12
13cc_library(
14    name = "arm_compute_core",
15    srcs = glob(
16        [
17            "src/core/*.cpp",
18            "src/core/helpers/*.cpp",
19            "src/core/CPP/**/*.cpp",
20            "src/core/utils/**/*.cpp",
21            "src/core/NEON/kernels/**/*.cpp",
22            "src/core/cpu/kernels/*.cpp",
23            "src/core/cpu/kernels/**/*.cpp",
24            "src/core/**/*.hpp",
25            "**/*.h",
26        ],
27        exclude = ["src/core/TracePoint.cpp"],
28    ),
29    hdrs = glob([
30        "arm_compute/core/**/*.h",
31        "**/*.inl",
32    ]) + [
33        "arm_compute_version.embed",
34    ],
35    defines = [
36        "ENABLE_FP16_KERNELS",
37        "ENABLE_FP32_KERNELS",
38        "ENABLE_QASYMM8_KERNELS",
39        "ENABLE_QASYMM8_SIGNED_KERNELS",
40        "ENABLE_QSYMM16_KERNELS",
41        "ENABLE_INTEGER_KERNELS",
42        "ENABLE_NHWC_KERNELS",
43    ],
44    includes = [
45        "arm_compute/runtime",
46        "src/core/NEON/kernels/assembly",
47        "src/core/NEON/kernels/convolution/common",
48        "src/core/NEON/kernels/convolution/winograd",
49    ],
50    deps = ["include"],
51)
52
53cc_library(
54    name = "arm_compute_runtime",
55    srcs = glob([
56        "src/runtime/*.cpp",
57        "src/runtime/CPP/**/*.cpp",
58        "src/runtime/NEON/**/*.cpp",
59        "src/runtime/cpu/**/*.cpp",
60        "**/*.h",
61    ]),
62    hdrs = glob([
63        "arm_compute/runtime/**/*.h",
64        "arm_compute/runtime/*.h",
65    ]) + [
66        "arm_compute_version.embed",
67    ],
68    defines = ["ARM_COMPUTE_CPP_SCHEDULER"],
69    linkopts = ["-lpthread"],
70    visibility = ["//visibility:public"],
71    deps = ["arm_compute_core"],
72)
73
74cc_library(
75    name = "arm_compute_graph",
76    srcs = glob([
77        "src/graph/*.cpp",
78        "src/graph/algorithms/*.cpp",
79        "src/graph/backends/*.cpp",
80        "src/graph/detail/*.cpp",
81        "src/graph/frontend/*.cpp",
82        "src/graph/mutators/*.cpp",
83        "src/graph/nodes/*.cpp",
84        "src/graph/printers/*.cpp",
85        "src/graph/backends/NEON/*.cpp",
86        "**/*.h",
87    ]),
88    hdrs = glob(["arm_compute/graph/**/*.h"]),
89    visibility = ["//visibility:public"],
90    deps = ["arm_compute_core"],
91)
92