Simplify blendZero for ARM64

8e3f5bc6 · Marc Marí · 5d755e7f · 8e3f5bc6
Commit 8e3f5bc6 authored May 02, 2020 by Marc Marí
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

openmmapi/include/openmm/internal/vectorize_neon.h openmmapi/include/openmm/internal/vectorize_neon.h +1 -1

No files found.
--- a/openmmapi/include/openmm/internal/vectorize_neon.h
+++ b/openmmapi/include/openmm/internal/vectorize_neon.h
@@ -427,7 +427,7 @@ static inline fvec4 blend(fvec4 v1, fvec4 v2, ivec4 mask) {
 }

 static inline fvec4 blendZero(fvec4 v, ivec4 mask) {
-    return blend(0.0f, v, mask);
+    return vreinterpretq_f32_s32(vandq_s32(vreinterpretq_s32_f32(v), mask));
 }

 static inline ivec4 blendZero(ivec4 v, ivec4 mask) {