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

Add a conv1d benchmark based on the whisper sizes. (#377)

Browse Source 
* Add a conv1d benchmark based on the whisper sizes.

* Enforce the batch-dim in conv1d.
This commit is contained in:
Laurent Mazare
2023-08-09 21:27:03 +02:00
committed by GitHub
parent 653ec5abc1
commit fcfdcbd337
4 changed files with 29 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,6 +1,6 @@
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ParamsConv1D {
pub(crate) b_size: Option<usize>,
pub(crate) b_size: usize,
// Maybe we should have a version without l_in as this bit depends on the input and not only on
// the weights.
pub(crate) l_in: usize,
@ -19,10 +19,7 @@ impl ParamsConv1D {
pub(crate) fn out_dims(&self) -> Vec<usize> {
let l_out = self.l_out();
match self.b_size {
None => vec![self.c_out, l_out],
Some(n) => vec![n, self.c_out, l_out],
}
vec![self.b_size, self.c_out, l_out]
}
}