diff options
| author | David Svantesson <david.svantesson@arm.com> | 2023-10-10 11:52:51 +0000 |
|---|---|---|
| committer | David Svantesson <david.svantesson@arm.com> | 2023-10-10 12:58:49 +0000 |
| commit | dfd56a630f7981100078b62adc37d65e48bc6d08 (patch) | |
| tree | 2e44554bc7dd5c5024ae5a0c801722e8516e9e2c /src/core/NEON/kernels | |
| parent | d9c1d44185e4fea72ce4d37d76021237fe64eeaf (diff) | |
| download | ComputeLibrary-dfd56a630f7981100078b62adc37d65e48bc6d08.tar.gz | |
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 <david.svantesson@arm.com>
Change-Id: I287763a547d2ab303c3f8ce10164178b680e551e
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10464
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels')
| -rw-r--r-- | src/core/NEON/kernels/NEReorderKernel.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
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; |