summaryrefslogtreecommitdiff
path: root/candle-core/src/cuda_backend.rs
diff options
context:
space:
mode:
authorMatt <Rocketknight1@users.noreply.github.com>2023-08-10 00:19:20 +0100
committerGitHub <noreply@github.com>2023-08-10 00:19:20 +0100
commit0dc1e5f387f91ff86cc8a4c09d5668e8baaab1b3 (patch)
tree922a5f1387c42b6101c98749211d5763529453d1 /candle-core/src/cuda_backend.rs
parent0cef3998fde542b9721215b77a80676a434b437f (diff)
parent25ec2d9f6bf36ff51c04f54f6c243828f6f4a8da (diff)
downloadcandle-0dc1e5f387f91ff86cc8a4c09d5668e8baaab1b3.tar.gz
candle-0dc1e5f387f91ff86cc8a4c09d5668e8baaab1b3.tar.bz2
candle-0dc1e5f387f91ff86cc8a4c09d5668e8baaab1b3.zip
Merge branch 'main' into readme_fixes
Diffstat (limited to 'candle-core/src/cuda_backend.rs')
-rw-r--r--candle-core/src/cuda_backend.rs2
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/cuda_backend.rs b/candle-core/src/cuda_backend.rs
index e51cc05d..a7f63353 100644
--- a/candle-core/src/cuda_backend.rs
+++ b/candle-core/src/cuda_backend.rs
@@ -904,7 +904,7 @@ impl<'a> Map2 for Conv1D<'a> {
let dims = shape.dims();
let el = shape.elem_count();
let l_out = p.l_out();
- let dst_el = p.c_out * l_out * p.b_size.unwrap_or(1);
+ let dst_el = p.c_out * l_out * p.b_size;
let cfg = LaunchConfig::for_num_elems(dst_el as u32);
let func = dev.get_or_load_func(&kernel_name::<T>("conv1d"), kernels::CONV)?;
// SAFETY: Set later by running the kernel.