fix: implement Vec::round fallback for armv7

OpenNMT · Feb 26, 2024 · ac99886 · ac99886
1 parent 42c6b96
commit ac99886
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/cpu/vec_neon.h b/src/cpu/vec_neon.h
@@ -160,7 +160,12 @@ namespace ctranslate2 {
       }
 
       static inline value_type round(value_type v) {
+#ifdef __aarch64__
         return vrndiq_f32(v);
+#else
+        float temp[4] = {std::nearbyintf(v[0]), std::nearbyintf(v[1]), std::nearbyintf(v[2]), std::nearbyintf(v[3])};
+        return load(temp);
+#endif
       }
 
       static inline void convert_and_store(value_type v, int8_t *a, dim_t count) {