summaryrefslogtreecommitdiff
path: root/candle-core/benches/matmul.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-core/benches/matmul.rs')
-rw-r--r--candle-core/benches/matmul.rs5
1 files changed, 2 insertions, 3 deletions
diff --git a/candle-core/benches/matmul.rs b/candle-core/benches/matmul.rs
index 4f7dfa6c..a5dba9cc 100644
--- a/candle-core/benches/matmul.rs
+++ b/candle-core/benches/matmul.rs
@@ -1,10 +1,9 @@
-mod bench_utils;
+mod utils;
-use crate::bench_utils::bench_name;
-use bench_utils::{device, BenchDevice};
use candle_core::{DType, Tensor};
use criterion::{black_box, criterion_group, criterion_main, Criterion, Throughput};
use std::time::Instant;
+use utils::{bench_name, device, BenchDevice};
fn run(a: &Tensor, b: &Tensor) {
a.matmul(&b.t().unwrap()).unwrap();