diff --git a/candle-core/src/conv.rs b/candle-core/src/conv.rs index 3ec7daa4..d544f5a2 100644 --- a/candle-core/src/conv.rs +++ b/candle-core/src/conv.rs @@ -175,7 +175,7 @@ impl Tensor { padding, stride, dilation, - cudnn_fwd_algo: None, + cudnn_fwd_algo: Some(CudnnFwdAlgo::ImplicitGemm), }; if groups == 1 { self.conv1d_single_group(kernel, ¶ms)