From 8e62723b2dbf7d5eda19d650c12ae4ae39678013 Mon Sep 17 00:00:00 2001 From: laurent Date: Sun, 13 Apr 2025 20:58:18 +0200 Subject: [PATCH] Set the algo. --- candle-core/src/conv.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)