missing bench

sonos · Oct 5, 2023 · 7a2adf2 · 7a2adf2
1 parent d8b0a29
commit 7a2adf2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/linalg/benches/leaky_relu.rs b/linalg/benches/leaky_relu.rs
@@ -12,6 +12,7 @@ fn leaky_relu_f16(c: &mut Criterion) {
     group.bench_function("rust", |b| b.iter(|| rust_fp16(input, alpha)));
     group.bench_function("rust_with_f16", |b| b.iter(|| unsafe { rust_with_fp16(input, alpha) }));
     group.bench_function("linalg", |b| b.iter(|| linalg16(input, alpha)));
+    group.bench_function("linalg-asm", |b| b.iter(|| tract_linalg::arm64::arm64fp16_leaky_relu_f16_16n::run(input, alpha)));
 }
 
 #[inline(never)]