add neon vector classifier bench

chengsun · May 29, 2022 · e75e8ec · e75e8ec
1 parent 51de759
commit e75e8ec
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/benches/benches.rs b/benches/benches.rs
@@ -302,6 +302,19 @@ fn bench_vector_classifier(c: &mut Criterion) {
         }
     }
 
+    #[cfg(target_arch = "aarch64")]
+    match vector_classifier::NeonBuilder::new() {
+        None => (),
+        Some(neon_builder) => {
+            let neon_classifier = neon_builder.build(&lookup_tables);
+            group.bench_function("neon",
+                                 |b| b.iter_batched(|| bytes.clone(), |mut bytes| {
+                                     neon_classifier.classify(&mut bytes);
+                                     black_box(bytes);
+                                 }, BatchSize::SmallInput));
+        }
+    }
+
     group.finish();
 }