summaryrefslogtreecommitdiff
path: root/candle-nn/examples/cpu_benchmarks.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-nn/examples/cpu_benchmarks.rs')
-rw-r--r--candle-nn/examples/cpu_benchmarks.rs2
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;