commit 26cda6d92094ffe3d1bc9114a560640351d610da
author Marat Dukhan <maratek@google.com> Thu Jan 09 13:54:32 2020 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Thu Jan 09 13:55:04 2020 -0800
tree a266fcec3524a2cd7e20889086cbb76629af550c
parent 6a6a421c251453af9c391bc857e95781a3ff541b

Fix gcc incompatibilities in NEON Sigmoid micro-kernels

PiperOrigin-RevId: 288965478

src/math/sigmoid-neonfma-rr2-p5-nr2recps.c

diff --git a/src/math/sigmoid-neonfma-rr2-p5-nr2recps.c b/src/math/sigmoid-neonfma-rr2-p5-nr2recps.c
index e388642..4a38969 100644
--- a/src/math/sigmoid-neonfma-rr2-p5-nr2recps.c
+++ b/src/math/sigmoid-neonfma-rr2-p5-nr2recps.c
@@ -97,7 +97,7 @@
     vf = vreinterpretq_f32_u32(vbicq_u32(vreinterpretq_u32_f32(vf), vcagtq_f32(vx, vdenorm_cutoff)));
 
     // Reconstruct sigmoid(x) = x < 0 ? sigmoid(-z) : 1.0 - sigmoid(-z)
-    const uint32x4_t vm = vcltq_s32(vx, vmovq_n_f32(0.0f));
+    const uint32x4_t vm = vcltq_f32(vx, vmovq_n_f32(0.0f));
     vf = vbslq_f32(vm, vf, vsubq_f32(vone, vf));
 
     vst1q_f32(output, vf); output += 4;