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 9ded5f71..68d384a6 100644 --- a/candle-nn/examples/cpu_benchmarks.rs +++ b/candle-nn/examples/cpu_benchmarks.rs @@ -6,7 +6,7 @@ extern crate intel_mkl_src; extern crate accelerate_src; use candle::quantized::GgmlType; -use candle::{CpuStorage, Device, Layout, Result, Shape, Tensor, D}; +use candle::{CpuStorage, Device, Layout, Module, Result, Shape, Tensor, D}; use clap::{Parser, Subcommand}; const CHECK_CONV2D: bool = false; |