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>
diff --git a/BUILD.bazel b/BUILD.bazel
index e3ad75a..d246336 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -72,6 +72,12 @@
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(
+ name = "sve_validation_flag",
+ flag_values = {
+ ":enable_sve_validation": "true",
+ },
+)
+
#---------------------------------------------------------------------
# Common defines used for all targets
@@ -138,8 +151,6 @@
"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 @@
"//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 @@
"//conditions:default": [],
}),
local_defines = [
+ "ENABLE_SVE",
+ "ARM_COMPUTE_ENABLE_SVE",
"ARM_COMPUTE_ENABLE_BF16",
],
deps = [
@@ -348,6 +363,8 @@
"//conditions:default": [],
}),
local_defines = [
+ "ENABLE_SVE",
+ "ARM_COMPUTE_ENABLE_SVE",
"ARM_COMPUTE_ENABLE_BF16",
],
visibility = ["//visibility:public"],