diff options
Diffstat (limited to 'candle-core/benches/benchmarks/qmatmul.rs')
| -rw-r--r-- | candle-core/benches/benchmarks/qmatmul.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-core/benches/benchmarks/qmatmul.rs b/candle-core/benches/benchmarks/qmatmul.rs index ccb136ac..4d34588b 100644 --- a/candle-core/benches/benchmarks/qmatmul.rs +++ b/candle-core/benches/benchmarks/qmatmul.rs @@ -7,7 +7,7 @@ use criterion::{black_box, criterion_group, Criterion, Throughput}; use std::time::Instant; fn run(matmul: &QMatMul, x: &Tensor) { - matmul.forward(&x).unwrap(); + matmul.forward(x).unwrap(); } fn run_bench(c: &mut Criterion, device: &Device, dtype: GgmlDType) { @@ -50,7 +50,7 @@ fn run_bench(c: &mut Criterion, device: &Device, dtype: GgmlDType) { fn criterion_benchmark(c: &mut Criterion) { let handler = BenchDeviceHandler::new().unwrap(); for device in handler.devices { - for dtype in vec![ + for dtype in [ GgmlDType::F32, GgmlDType::F16, GgmlDType::Q4_0, |