diff options
Diffstat (limited to 'BUILD.bazel')
| -rw-r--r-- | BUILD.bazel | 414 |
1 files changed, 414 insertions, 0 deletions
diff --git a/BUILD.bazel b/BUILD.bazel new file mode 100644 index 0000000000..3a7d941a0e --- /dev/null +++ b/BUILD.bazel @@ -0,0 +1,414 @@ +# Copyright (c) 2023 Arm Limited. +# +# SPDX-License-Identifier: MIT +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the "Software"), to +# deal in the Software without restriction, including without limitation the +# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or +# sell copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in all +# copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +# SOFTWARE. + +load("@bazel_skylib//rules:common_settings.bzl", "string_flag") +load("@bazel_skylib//rules:common_settings.bzl", "bool_flag") + +#--------------------------------------------------------------------- +# Config setting for Tensorflow build +config_setting( + name = "build_with_acl", + define_values = { + "build_with_acl": "true", + }, + visibility = ["//visibility:public"], +) + +#--------------------------------------------------------------------- +# Flags for build options. Example: --//:debug=true +# All flags have aliases in .bazelrc so can use for example --debug=true when building +bool_flag( + name = "debug", + build_setting_default = False, + visibility = ["//visibility:public"], +) + +bool_flag( + name = "Werror", + build_setting_default = False, + visibility = ["//visibility:public"], +) + +bool_flag( + name = "logging", + build_setting_default = False, + visibility = ["//visibility:public"], +) + +bool_flag( + name = "openmp", + build_setting_default = True, + visibility = ["//visibility:public"], +) + +bool_flag( + name = "cppthreads", + build_setting_default = False, + visibility = ["//visibility:public"], +) + +bool_flag( + name = "enable_bf16_validation", + build_setting_default = False, + visibility = ["//visibility:public"], +) + +bool_flag( + name = "enable_sve_validation", + build_setting_default = False, + visibility = ["//visibility:public"], +) + +string_flag( + name = "arch", + build_setting_default = "armv8-a", + values = [ + "armv8-a", + "armv8.2-a+fp16" + ] +) + +#--------------------------------------------------------------------- +# Flag variables +config_setting( + name = "debug_flag", + flag_values = { + ":debug": "true", + }, +) + +config_setting( + name = "Werror_flag", + flag_values = { + ":Werror": "true", + }, +) + +config_setting( + name = "logging_flag", + flag_values = { + ":logging": "true", + }, +) + +config_setting( + name = "openmp_flag", + flag_values = { + ":openmp": "true", + }, +) + +config_setting( + name = "cppthreads_flag", + flag_values = { + ":cppthreads": "true", + }, +) + +config_setting( + name = "bf16_validation_flag", + flag_values = { + ":enable_bf16_validation": "true", + }, +) + +config_setting( + name = "sve_validation_flag", + flag_values = { + ":enable_sve_validation": "true", + }, +) + +config_setting( + name = "arch_armv8-a", + flag_values = { + "arch": "armv8-a" + } +) + +config_setting( + name = "arch_armv8.2-a+fp16", + flag_values = { + "arch": "armv8.2-a+fp16" + } +) + + +#--------------------------------------------------------------------- +# Common defines used for all targets +cc_library( + name = "common_defines", + defines = [ + "ENABLE_NEON", + "ARM_COMPUTE_CPU_ENABLED", + "ARM_COMPUTE_ENABLE_NEON", + "ARM_COMPUTE_ENABLE_I8MM", + "ENABLE_FP32_KERNELS", + "ENABLE_QASYMM8_KERNELS", + "ENABLE_QASYMM8_SIGNED_KERNELS", + "ENABLE_QSYMM16_KERNELS", + "ENABLE_INTEGER_KERNELS", + "ENABLE_NHWC_KERNELS", + "ENABLE_NCHW_KERNELS", + "ARM_COMPUTE_GRAPH_ENABLED", + "ARM_COMPUTE_ENABLE_SVEF32MM", + "ARM_COMPUTE_ENABLE_FIXED_FORMAT_KERNELS", + "_GLIBCXX_USE_NANOSLEEP" + ] + select({ + "//:debug_flag": [ + "ARM_COMPUTE_DEBUG_ENABLED", + "ARM_COMPUTE_ASSERTS_ENABLED", + ], + "//conditions:default": [], + }) + + select({ + "//:logging_flag": ["ARM_COMPUTE_LOGGING_ENABLED"], + "//conditions:default": [], + }) + + select({ + "//:cppthreads_flag": ["ARM_COMPUTE_CPP_SCHEDULER"], + "//conditions:default": [], + }) + + select({ + "//:openmp_flag": ["ARM_COMPUTE_OPENMP_SCHEDULER"], + "//conditions:default": [], + }) + + select({ + "//:arch_armv8-a": [], + "//:arch_armv8.2-a+fp16": ["ENABLE_FP16_KERNELS", "ARM_COMPUTE_ENABLE_FP16"], + "//conditions:default": [], + }), + visibility = ["//visibility:public"], +) + +#--------------------------------------------------------------------- +# Rule for creating file "arm_compute_version.embed" +genrule( + name = "create_version_file", + srcs = ["SConscript"], + outs = ["arm_compute_version.embed"], + cmd = "$(location //scripts:print_version_file) 'n/a' 'n/a' 'true' > $@", + tools = ["//scripts:print_version_file"], + visibility = ["//visibility:public"], +) + +#--------------------------------------------------------------------- +# Graph library + +cc_library( + name = "arm_compute_graph", + srcs = ["//src:arm_compute_graph_srcs"], + copts = [] + select({ + "//:arch_armv8-a": ["-march=armv8-a"], + "//:arch_armv8.2-a+fp16": ["-march=armv8.2-a+fp16"], + "//conditions:default": ["-march=armv8-a"], + }) + select({ + "//:debug_flag": [ + "-O0", + "-g", + "-gdwarf-2", + ], + "//conditions:default": ["-O3"], + }) + + select({ + "//:openmp_flag": ["-fopenmp"], + "//conditions:default": [], + }) + + select({ + "//:Werror_flag": ["-Werror"], + "//conditions:default": [], + }), + visibility = ["//visibility:public"], + deps = [ + "arm_compute", + "//:common_defines", + "//arm_compute:graph_headers", + ], + alwayslink = True, +) + +#--------------------------------------------------------------------- +# SVE2 library + +cc_library( + name = "arm_compute_sve2", + srcs = ["//src:arm_compute_sve2_srcs"], + copts = [ + "-march=armv8.6-a+sve2+fp16+dotprod", + ] + select({ + "//:debug_flag": [ + "-O0", + "-g", + "-gdwarf-2", + ], + "//conditions:default": ["-O3"], + }) + + select({ + "//:openmp_flag": ["-fopenmp"], + "//conditions:default": [], + }) + + select({ + "//:Werror_flag": ["-Werror"], + "//conditions:default": [], + }), + includes = [ + "src/core/NEON/kernels/arm_conv", + "src/core/NEON/kernels/arm_gemm", + "src/core/NEON/kernels/assembly", + "src/core/cpu/kernels/assembly", + "src/cpu/kernels/assembly", + ], + linkopts = select({ + "//:openmp_flag": ["-fopenmp"], + "//conditions:default": [], + }), + local_defines = [ + "ENABLE_SVE", + "ARM_COMPUTE_ENABLE_SVE", + "ARM_COMPUTE_ENABLE_SVE2", + "ARM_COMPUTE_ENABLE_BF16", + ], + deps = [ + "//:common_defines", + "//arm_compute:core_headers", + "//arm_compute:runtime_headers", + "//include", + "//support", + ], + alwayslink = True, +) + +#--------------------------------------------------------------------- +# SVE library + +cc_library( + name = "arm_compute_sve", + srcs = ["//src:arm_compute_sve_srcs"], + copts = [ + "-march=armv8.2-a+sve+fp16+dotprod", + ] + select({ + "//:debug_flag": [ + "-O0", + "-g", + "-gdwarf-2", + ], + "//conditions:default": ["-O3"], + }) + + select({ + "//:openmp_flag": ["-fopenmp"], + "//conditions:default": [], + }) + + select({ + "//:Werror_flag": ["-Werror"], + "//conditions:default": [], + }), + includes = [ + "src/core/NEON/kernels/arm_conv", + "src/core/NEON/kernels/arm_gemm", + "src/core/NEON/kernels/assembly", + "src/core/cpu/kernels/assembly", + "src/cpu/kernels/assembly", + ], + linkopts = select({ + "//:openmp_flag": ["-fopenmp"], + "//conditions:default": [], + }), + local_defines = [ + "ENABLE_SVE", + "ARM_COMPUTE_ENABLE_SVE", + "ARM_COMPUTE_ENABLE_BF16", + ], + deps = [ + "//:common_defines", + "//arm_compute:core_headers", + "//arm_compute:runtime_headers", + "//include", + "//support", + ], + alwayslink = True, +) + +#--------------------------------------------------------------------- +# Core and Runtime library + +cc_library( + name = "arm_compute", + srcs = ["//src:arm_compute_srcs"], + hdrs = glob([ + "core/NEON/kernels/**/*.h", + "core/NEON/kernels/**/*.hpp", + "**/*.inl", + ]) + [ + "//:create_version_file", + ], + copts = [] + select({ + "//:arch_armv8-a": ["-march=armv8-a"], + "//:arch_armv8.2-a+fp16": ["-march=armv8.2-a+fp16"], + "//conditions:default": ["-march=armv8-a"], + }) + select({ + "//:debug_flag": [ + "-O0", + "-g", + "-gdwarf-2", + ], + "//conditions:default": ["-O3"], + }) + + select({ + "//:openmp_flag": ["-fopenmp"], + "//conditions:default": [], + }) + + select({ + "//:Werror_flag": ["-Werror"], + "//conditions:default": [], + }), + includes = [ + "arm_compute/runtime", + "src/core/NEON/kernels/arm_gemm", + "src/core/NEON/kernels/assembly", + "src/core/NEON/kernels/convolution/common", + "src/core/NEON/kernels/convolution/winograd", + "src/core/cpu/kernels/assembly", + "src/cpu/kernels/assembly", + ], + linkopts = select({ + "//:openmp_flag": ["-fopenmp"], + "//conditions:default": [], + }), + local_defines = [ + "ENABLE_SVE", + "ARM_COMPUTE_ENABLE_SVE", + "ARM_COMPUTE_ENABLE_BF16", + ], + visibility = ["//visibility:public"], + deps = [ + "//:common_defines", + "//arm_compute:core_headers", + "//arm_compute:graph_headers", + "//arm_compute:runtime_headers", + "//include", + "//support", + "//utils", + "//:arm_compute_sve", + "//:arm_compute_sve2" + ], + alwayslink = True, +) |