From aba1e90797e430f28eec13b14b76dd5355876f9c Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Wed, 23 Aug 2023 12:58:55 +0100 Subject: 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. --- candle-core/examples/cpu_benchmarks.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'candle-core/examples/cpu_benchmarks.rs') 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 { - 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 { - d.0.conv2d(&d.1, 0, 1) + d.0.conv2d(&d.1, 0, 1, 1) } const ITERS: usize = 1; -- cgit v1.2.3