92 lines
2.3 KiB
Python
92 lines
2.3 KiB
Python
exports_files(["LICENSE"])
|
|
|
|
cc_library(
|
|
name = "include",
|
|
hdrs = glob([
|
|
"include/**/*.h",
|
|
"include/**/*.hpp",
|
|
]),
|
|
includes = ["include"],
|
|
strip_include_prefix = "include",
|
|
)
|
|
|
|
cc_library(
|
|
name = "arm_compute_core",
|
|
srcs = glob(
|
|
[
|
|
"src/core/*.cpp",
|
|
"src/core/helpers/*.cpp",
|
|
"src/core/CPP/**/*.cpp",
|
|
"src/core/utils/**/*.cpp",
|
|
"src/core/NEON/kernels/**/*.cpp",
|
|
"src/core/cpu/kernels/*.cpp",
|
|
"src/core/cpu/kernels/**/*.cpp",
|
|
"src/core/**/*.hpp",
|
|
"**/*.h",
|
|
],
|
|
exclude = ["src/core/TracePoint.cpp"],
|
|
),
|
|
hdrs = glob([
|
|
"arm_compute/core/**/*.h",
|
|
"**/*.inl",
|
|
]) + [
|
|
"arm_compute_version.embed",
|
|
],
|
|
defines = [
|
|
"ENABLE_FP16_KERNELS",
|
|
"ENABLE_FP32_KERNELS",
|
|
"ENABLE_QASYMM8_KERNELS",
|
|
"ENABLE_QASYMM8_SIGNED_KERNELS",
|
|
"ENABLE_QSYMM16_KERNELS",
|
|
"ENABLE_INTEGER_KERNELS",
|
|
"ENABLE_NHWC_KERNELS",
|
|
],
|
|
includes = [
|
|
"arm_compute/runtime",
|
|
"src/core/NEON/kernels/assembly",
|
|
"src/core/NEON/kernels/convolution/common",
|
|
"src/core/NEON/kernels/convolution/winograd",
|
|
],
|
|
deps = ["include"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "arm_compute_runtime",
|
|
srcs = glob([
|
|
"src/runtime/*.cpp",
|
|
"src/runtime/CPP/**/*.cpp",
|
|
"src/runtime/NEON/**/*.cpp",
|
|
"src/runtime/cpu/**/*.cpp",
|
|
"**/*.h",
|
|
]),
|
|
hdrs = glob([
|
|
"arm_compute/runtime/**/*.h",
|
|
"arm_compute/runtime/*.h",
|
|
]) + [
|
|
"arm_compute_version.embed",
|
|
],
|
|
defines = ["ARM_COMPUTE_CPP_SCHEDULER"],
|
|
linkopts = ["-lpthread"],
|
|
visibility = ["//visibility:public"],
|
|
deps = ["arm_compute_core"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "arm_compute_graph",
|
|
srcs = glob([
|
|
"src/graph/*.cpp",
|
|
"src/graph/algorithms/*.cpp",
|
|
"src/graph/backends/*.cpp",
|
|
"src/graph/detail/*.cpp",
|
|
"src/graph/frontend/*.cpp",
|
|
"src/graph/mutators/*.cpp",
|
|
"src/graph/nodes/*.cpp",
|
|
"src/graph/printers/*.cpp",
|
|
"src/graph/backends/NEON/*.cpp",
|
|
"**/*.h",
|
|
]),
|
|
hdrs = glob(["arm_compute/graph/**/*.h"]),
|
|
visibility = ["//visibility:public"],
|
|
deps = ["arm_compute_core"],
|
|
)
|