commit c72fa1e65f314612a9746a6ce783c0b218695f1b
author Marat Dukhan <maratek@google.com> Wed Nov 27 11:54:03 2019 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Wed Nov 27 11:54:31 2019 -0800
tree e17b227f72a59a1a50c11f33fec29a1db750497c
parent 69172d92ecd316804cc8a80e779b07cc51defb81

Use XNN_ARCH_* macros for architecture-specific parts in micro-kernels

PiperOrigin-RevId: 282810001

src/q8-igemm/4x8-neon.c

diff --git a/src/q8-igemm/4x8-neon.c b/src/q8-igemm/4x8-neon.c
index e7ad227..036258e 100644
--- a/src/q8-igemm/4x8-neon.c
+++ b/src/q8-igemm/4x8-neon.c
@@ -10,6 +10,7 @@
 
 #include <arm_neon.h>
 
+#include <xnnpack/common.h>
 #include <xnnpack/igemm.h>
 
 
@@ -347,7 +348,7 @@
     vacc3x4567 = vrshlq_s32(vacc3x4567, vright_shift);
 
     const int16x8_t voutput_zero_point = vld1q_dup_s16(&params->neon.output_zero_point);
-  #ifdef __aarch64__
+  #if XNN_ARCH_ARM64
     const int16x8_t vacc0x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc0x0123), vacc0x4567), voutput_zero_point);
     const int16x8_t vacc1x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc1x0123), vacc1x4567), voutput_zero_point);
     const int16x8_t vacc2x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc2x0123), vacc2x4567), voutput_zero_point);