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

Set the algo.

Browse Source
This commit is contained in:
laurent
2025-04-13 20:58:18 +02:00
parent fb660b8d43
commit 8e62723b2d
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/conv.rs
View File

@ -175,7 +175,7 @@ impl Tensor {
padding, padding,
stride, stride,
dilation, dilation,
cudnn_fwd_algo: None, cudnn_fwd_algo: Some(CudnnFwdAlgo::ImplicitGemm),
}; };
if groups == 1 { if groups == 1 {
self.conv1d_single_group(kernel, &params) self.conv1d_single_group(kernel, &params)