Update impl

Author: Brooooooklyn

Cargo.toml

@@ -13,7 +13,7 @@ name = "escape"
 path = "examples/escape.rs"
 
 [features]
-force_aarch64_generic = [] # Force use of generic implementation on aarch64
+force_aarch64_neon = [] # Force use of neon implementation on aarch64
 
 [[bench]]
 name = "escape"

README.md

@@ -10,7 +10,7 @@ Optimized SIMD routines for escaping JSON strings. This repository contains the
 
 > [!NOTE]
 >
-> The `force_aarch64_generic` feature flag can be used to force use of the generic fallback on aarch64. This is useful for testing the generic fallback on aarch64 devices with smaller register numbers.
+> The `force_aarch64_neon` feature flag can be used to force use of the neon implementation on aarch64. This is useful for the benchmark.
 
 ## Benchmarks
 

src/lib.rs

@@ -1,7 +1,7 @@
 #[cfg(target_arch = "x86_64")]
 mod x86;
 
-#[cfg(all(target_arch = "aarch64", not(feature = "force_aarch64_generic")))]
+#[cfg(target_arch = "aarch64")]
 mod aarch64;
 
 const BB: u8 = b'b'; // \x08
@@ -150,19 +150,17 @@ pub fn escape<S: AsRef<str>>(input: S) -> String {
 
     #[cfg(target_arch = "aarch64")]
     {
-        use std::arch::is_aarch64_feature_detected;
-
-        #[cfg(feature = "force_aarch64_generic")]
+        #[cfg(feature = "force_aarch64_neon")]
         {
-            return escape_generic(input);
+            return aarch64::escape_neon(input);
         }
-        #[cfg(not(feature = "force_aarch64_generic"))]
+        #[cfg(not(feature = "force_aarch64_neon"))]
         {
             // on Apple M2 and later, the `bf16` feature is available
             // it means they have more registers and can significantly benefit from the SIMD path
             // TODO: add support for sve2 chips with wider registers
             // github actions ubuntu-24.04-arm runner has 128 bits sve2 registers, it's not enough for the SIMD path
-            if cfg!(target_os = "macos") && is_aarch64_feature_detected!("bf16") {
+            if cfg!(target_os = "macos") && std::arch::is_aarch64_feature_detected!("bf16") {
                 return aarch64::escape_neon(input);
             } else {
                 return escape_generic(input);