From fd1104b1a173d95ae8ee4c37cce3f54304a0d9ac Mon Sep 17 00:00:00 2001
From: laurent <laurent.mazare@gmail.com>
Date: Mon, 18 Sep 2023 09:14:00 +0100
Subject: [PATCH] Bench fix.

---
 gemm/examples/gemv_bench.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gemm/examples/gemv_bench.rs b/gemm/examples/gemv_bench.rs
index cd3bcce..40549e8 100644
--- a/gemm/examples/gemv_bench.rs
+++ b/gemm/examples/gemv_bench.rs
@@ -6,7 +6,7 @@ pub fn run(m: usize, n: usize, k: usize, niters: usize) {
     let (lhs_rs, lhs_cs) = (k, 1);
     let (rhs_rs, rhs_cs) = (1, k);
     let mut run = move || unsafe {
-        gemm::gemm(
+        candle_gemm::gemm(
             m,
             n,
             k,
@@ -25,7 +25,7 @@ pub fn run(m: usize, n: usize, k: usize, niters: usize) {
             false,
             false,
             false,
-            gemm::Parallelism::None,
+            candle_gemm::Parallelism::None,
         )
     };
     for _i in 0..10 {