commit 548e83046d063f12b58662789f0f5b06019226ce
author Marat Dukhan <maratek@google.com> Tue Aug 04 15:03:46 2020 -0700
committer XNNPACK Team <xnnpack-github-robot@google.com> Tue Aug 04 15:04:19 2020 -0700
tree c8faad74893cabafedfa8338d0bc2220107c591e
parent f62bbdcab25864defdfa3085be911e32b6e10f3e

Fix NCW Global Average Pooling on WAsm SIMD

PiperOrigin-RevId: 324898342

src/xnnpack/params-init.h

diff --git a/src/xnnpack/params-init.h b/src/xnnpack/params-init.h
index 5a92191..255503a 100644
--- a/src/xnnpack/params-init.h
+++ b/src/xnnpack/params-init.h
@@ -491,9 +491,9 @@
 
     const uint32_t w = (width - 1) & 3;
     params->scalar.mask[0] = UINT32_C(0xFFFFFFFF);
-    params->scalar.mask[1] = (int32_t) (w >= 1);
-    params->scalar.mask[2] = (int32_t) (w >= 2);
-    params->scalar.mask[3] = (int32_t) (w >= 3);
+    params->scalar.mask[1] = -(int32_t) (w >= 1);
+    params->scalar.mask[2] = -(int32_t) (w >= 2);
+    params->scalar.mask[3] = -(int32_t) (w >= 3);
   #endif
 }