diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-23 12:58:55 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-08-23 12:58:55 +0100 |
commit | aba1e90797e430f28eec13b14b76dd5355876f9c (patch) | |
tree | 16bcf7fb151715d3bcdbec2b5263922bd0bdd35a /candle-core/examples/cpu_benchmarks.rs | |
parent | 4ee1cf038ada55ec477dcd6496cf2aec1902775b (diff) | |
download | candle-aba1e90797e430f28eec13b14b76dd5355876f9c.tar.gz candle-aba1e90797e430f28eec13b14b76dd5355876f9c.tar.bz2 candle-aba1e90797e430f28eec13b14b76dd5355876f9c.zip |
Add some group parameter to convolutions. (#566)
* Add some group parameter to convolutions.
* Avoid some unnecessary groups checks.
* Move the tensor convolution bits.
* Properh handling of groups.
* Bump the crate version.
* And add a changelog.
Diffstat (limited to 'candle-core/examples/cpu_benchmarks.rs')
-rw-r--r-- | candle-core/examples/cpu_benchmarks.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-core/examples/cpu_benchmarks.rs b/candle-core/examples/cpu_benchmarks.rs index d7f60f81..1ebd9b75 100644 --- a/candle-core/examples/cpu_benchmarks.rs +++ b/candle-core/examples/cpu_benchmarks.rs @@ -40,7 +40,7 @@ impl Benchmark for Conv1d { } fn run_one(d: &Self::PreProcessData) -> Result<Self::RunResult> { - d.0.conv1d(&d.1, 0, 1) + d.0.conv1d(&d.1, 0, 1, 1) } const ITERS: usize = 5; @@ -59,7 +59,7 @@ impl Benchmark for Conv2d { } fn run_one(d: &Self::PreProcessData) -> Result<Self::RunResult> { - d.0.conv2d(&d.1, 0, 1) + d.0.conv2d(&d.1, 0, 1, 1) } const ITERS: usize = 1; |