aboutsummaryrefslogtreecommitdiff
path: root/BUILD.bazel
diff options
context:
space:
mode:
authorDavid Svantesson <david.svantesson@arm.com>2023-03-28 14:13:32 +0000
committerDavid Svantesson <david.svantesson@arm.com>2023-04-28 16:07:43 +0000
commit3b162e53267d13d18891baf3372f971f1d4213d3 (patch)
tree529f0588165ce40753e7ee79f5933d79d57d30bf /BUILD.bazel
parenta25582c4e7dddd26419e0a3316614e8309928934 (diff)
downloadComputeLibrary-3b162e53267d13d18891baf3372f971f1d4213d3.tar.gz
Reorder added
Adds Reorder kernel exposing blocking reorders from arm_gemm Resolves ONCPUML-1232 Change-Id: I42bf4166311fe1771565134d3ed7039fc8e30230 Signed-off-by: David Svantesson <david.svantesson@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9500 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: SiCong Li <sicong.li@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'BUILD.bazel')
-rw-r--r--BUILD.bazel23
1 files changed, 20 insertions, 3 deletions
diff --git a/BUILD.bazel b/BUILD.bazel
index e3ad75abd8..d246336c29 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -72,6 +72,12 @@ bool_flag(
visibility = ["//visibility:public"],
)
+bool_flag(
+ name = "enable_sve_validation",
+ build_setting_default = False,
+ visibility = ["//visibility:public"],
+)
+
#---------------------------------------------------------------------
# Flag variables
config_setting(
@@ -116,6 +122,13 @@ config_setting(
},
)
+config_setting(
+ name = "sve_validation_flag",
+ flag_values = {
+ ":enable_sve_validation": "true",
+ },
+)
+
#---------------------------------------------------------------------
# Common defines used for all targets
@@ -138,8 +151,6 @@ cc_library(
"DARM_COMPUTE_GRAPH_ENABLED",
"ARM_COMPUTE_ENABLE_SVEF32MM",
"ARM_COMPUTE_ENABLE_FIXED_FORMAT_KERNELS",
- "ENABLE_SVE",
- "ARM_COMPUTE_ENABLE_SVE",
"_GLIBCXX_USE_NANOSLEEP"
] + select({
"//:debug_flag": [
@@ -243,8 +254,10 @@ cc_library(
"//conditions:default": [],
}),
local_defines = [
+ "ENABLE_SVE",
+ "ARM_COMPUTE_ENABLE_SVE",
"ARM_COMPUTE_ENABLE_SVE2",
- "ARM_COMPUTE_ENABLE_BF16"
+ "ARM_COMPUTE_ENABLE_BF16",
],
deps = [
"//:common_defines",
@@ -292,6 +305,8 @@ cc_library(
"//conditions:default": [],
}),
local_defines = [
+ "ENABLE_SVE",
+ "ARM_COMPUTE_ENABLE_SVE",
"ARM_COMPUTE_ENABLE_BF16",
],
deps = [
@@ -348,6 +363,8 @@ cc_library(
"//conditions:default": [],
}),
local_defines = [
+ "ENABLE_SVE",
+ "ARM_COMPUTE_ENABLE_SVE",
"ARM_COMPUTE_ENABLE_BF16",
],
visibility = ["//visibility:public"],