From 629f9f1c2a30e2bf3d41be2bb94d2618d16f829f Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9C=A4=EC=A7=80=EC=98=81/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84?= =?utf8?q?=EC=9E=90?= Date: Fri, 31 Aug 2018 12:55:34 +0900 Subject: [PATCH] Add assertion in ToARMCompute function (#2540) This commit prevents a segfault from wrong access. Signed-off-by: Jiyoung Yun --- runtimes/pure_arm_compute/src/internal/Swizzle.h | 1 + 1 file changed, 1 insertion(+) diff --git a/runtimes/pure_arm_compute/src/internal/Swizzle.h b/runtimes/pure_arm_compute/src/internal/Swizzle.h index 8d10099..ddf017f 100644 --- 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) -- 2.7.4