diff options
Diffstat (limited to 'candle-nn/examples/cpu_benchmarks.rs')
| -rw-r--r-- | candle-nn/examples/cpu_benchmarks.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/examples/cpu_benchmarks.rs b/candle-nn/examples/cpu_benchmarks.rs index 204a7109..e58ea727 100644 --- a/candle-nn/examples/cpu_benchmarks.rs +++ b/candle-nn/examples/cpu_benchmarks.rs @@ -206,7 +206,7 @@ impl Benchmark for QMatMul { fn preprocess() -> Result<Self::PreProcessData> { let zeros = vec![candle::quantized::k_quants::BlockQ4_0::zeros(); 4096 * 11008 / 32]; let mm = candle::quantized::QTensor::new(zeros, (4096, 11008))?; - let mm = candle::quantized::QMatMul::from_qtensor(mm); + let mm = candle::quantized::QMatMul::from_qtensor(mm)?; let arg = Tensor::randn(0f32, 1., (128, 11008), &Device::Cpu)?; Ok((mm, arg)) } |