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

Add more of the conv1d op.

Browse Source
This commit is contained in:
laurent
2023-07-04 11:15:45 +01:00
parent 3aac1047fe
commit a424d95473
8 changed files with 52 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-core/src/storage.rs
View File

@ -149,18 +149,17 @@ impl Storage {
l: &Layout,
kernel: &Self,
kernel_l: &Layout,
padding: usize,
stride: usize,
params: &crate::conv::ParamsConv1D,
) -> Result<Self> {
self.same_device(kernel, "conv1d")?;
self.same_dtype(kernel, "conv1d")?;
match (self, &kernel) {
(Storage::Cpu(inp), Storage::Cpu(kernel)) => {
let s = inp.conv1d(l, kernel, kernel_l, padding, stride)?;
let s = inp.conv1d(l, kernel, kernel_l, params)?;
Ok(Self::Cpu(s))
}
(Storage::Cuda(inp), Storage::Cuda(kernel)) => {
let s = inp.conv1d(l, kernel, kernel_l, padding, stride)?;
let s = inp.conv1d(l, kernel, kernel_l, params)?;
Ok(Self::Cuda(s))
}
(lhs, rhs) => Err(Error::DeviceMismatchBinaryOp {