huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 19:58:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: remove incorrect unwrap (#379)

Browse Source
This commit is contained in:
Ciarán Curley
2023-08-09 21:45:24 +01:00
committed by GitHub
parent da26e2832c
commit 25ec2d9f6b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/cuda_backend.rs
View File

@ -904,7 +904,7 @@ impl<'a> Map2 for Conv1D<'a> {
let dims = shape.dims(); let dims = shape.dims();
let el = shape.elem_count(); let el = shape.elem_count();
let l_out = p.l_out(); 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 cfg = LaunchConfig::for_num_elems(dst_el as u32);
let func = dev.get_or_load_func(&kernel_name::<T>("conv1d"), kernels::CONV)?; let func = dev.get_or_load_func(&kernel_name::<T>("conv1d"), kernels::CONV)?;
// SAFETY: Set later by running the kernel. // SAFETY: Set later by running the kernel.