Support for cudnn conv1d. (#2888)

* Support for cudnn conv1d.

* More conv1d work.

* Get the conv1d to work with cudnn.

* Cleanup.
This commit is contained in:
Laurent Mazare
2025-04-13 16:47:37 +02:00
committed by GitHub
parent b44d38de0e
commit f3a73f80d1
3 changed files with 172 additions and 0 deletions

View File

@ -14,6 +14,7 @@ pub struct ParamsConv1D {
pub(crate) padding: usize,
pub(crate) stride: usize,
pub(crate) dilation: usize,
pub(crate) cudnn_fwd_algo: Option<CudnnFwdAlgo>,
}
impl ParamsConv1D {
@ -174,6 +175,7 @@ impl Tensor {
padding,
stride,
dilation,
cudnn_fwd_algo: None,
};
if groups == 1 {
self.conv1d_single_group(kernel, &params)