From dfd56a630f7981100078b62adc37d65e48bc6d08 Mon Sep 17 00:00:00 2001 From: David Svantesson Date: Tue, 10 Oct 2023 11:52:51 +0000 Subject: Fix NEReorderKernel validation Partially resolves: ONCPUML-1251 Reorders of block size 8 are only available when SVE is enabled. This fixes an issue with these reorders being checked even when SVE is not enabled. Signed-off-by: David Svantesson Change-Id: I287763a547d2ab303c3f8ce10164178b680e551e Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10464 Benchmark: Arm Jenkins Reviewed-by: Viet-Hoa Do Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/core/NEON/kernels/NEReorderKernel.cpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/core/NEON/kernels') diff --git a/src/core/NEON/kernels/NEReorderKernel.cpp b/src/core/NEON/kernels/NEReorderKernel.cpp index f92a4c87da..6c2c987eb7 100644 --- a/src/core/NEON/kernels/NEReorderKernel.cpp +++ b/src/core/NEON/kernels/NEReorderKernel.cpp @@ -211,11 +211,13 @@ Status NEReorderKernel::validate(const ITensorInfo *input, int ksize; switch (output_wf) { +#if defined(ARM_COMPUTE_ENABLE_SVE) case WeightFormat::OHWIo8: { ksize = 8; break; } +#endif /* ARM_COMPUTE_ENABLE_SVE */ case WeightFormat::OHWIo4: { ksize = 4; -- cgit v1.2.1