diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-11-25 10:09:45 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-25 10:09:45 +0000 |
| commit | bfa7c8fc01cd92896d404d528128dbac7c2b8b9c (patch) | |
| tree | a00f281b2d6b963c208c1182b647eef604f373ae /candle-nn | |
| parent | 762e996ce636c5168ec1dc3ded9bd729d4f14d84 (diff) | |
| download | candle-bfa7c8fc01cd92896d404d528128dbac7c2b8b9c.tar.gz candle-bfa7c8fc01cd92896d404d528128dbac7c2b8b9c.tar.bz2 candle-bfa7c8fc01cd92896d404d528128dbac7c2b8b9c.zip | |
Implement the module trait directly for QMatMul. (#1372)
Diffstat (limited to 'candle-nn')
| -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; |