Add assertion in ToARMCompute function (#2540)

This commit prevents a segfault from wrong access.

Signed-off-by: Jiyoung Yun <jy910.yun@samsung.com>

diff --git a/runtimes/pure_arm_compute/src/internal/Swizzle.h b/runtimes/pure_arm_compute/src/internal/Swizzle.h
index 8d10099..ddf017f 100644 (file)
--- a/runtimes/pure_arm_compute/src/internal/Swizzle.h
+++ b/runtimes/pure_arm_compute/src/internal/Swizzle.h
@@ -22,6 +22,7 @@ private:
 // Convert T/F Lite / NNAPI axis (based on ...NHWC) to ARMCompute axis (WHCN...)
 inline ARMComputeAxis ToARMComputeAxis(uint32_t rank, uint32_t axis)
 {
+  assert(rank > axis);
   const ARMComputeAxis reversed{(rank - axis) - 1};
 
   if (rank < 4)