diff options
Diffstat (limited to 'candle-core/src/cudnn.rs')
| -rw-r--r-- | candle-core/src/cudnn.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/cudnn.rs b/candle-core/src/cudnn.rs index 3e943e51..235ad6e3 100644 --- a/candle-core/src/cudnn.rs +++ b/candle-core/src/cudnn.rs @@ -48,7 +48,7 @@ pub(crate) fn launch_conv2d< let conv = cudnn.create_conv2d::<T>( /* pad */ [params.padding as i32, params.padding as i32], /* stride */ [params.stride as i32, params.stride as i32], - /* dilation */ [1, 1], + /* dilation */ [params.dilation as i32, params.dilation as i32], cudarc::cudnn::sys::cudnnConvolutionMode_t::CUDNN_CROSS_CORRELATION, )?; let x_shape = [ |