Bench fix.

LaurentMazare · Sep 18, 2023 · fd1104b · fd1104b
1 parent 92adfe4
commit fd1104b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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 {