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

Bugfix for the conv2d cpu kernel. (#820)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-11 23:11:27 +01:00
committed by GitHub
parent c5a058b169
commit 871efc0307
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/cpu_backend.rs
View File

@ -1209,7 +1209,7 @@ impl Map1 for Im2Col {
let dst_idx = dst_idx + h_k_idx * w_k;
for w_k_idx in 0..w_k {
let src_w = w_idx * stride + w_k_idx * dilation;
if padding != 0 && (src_w < padding || src_w >= h + padding) {
if padding != 0 && (src_w < padding || src_w >= w + padding) {
continue;
}
let src_w = src_w - padding;

2
candle-nn/examples/cpu_benchmarks.rs
View File

@ -84,7 +84,7 @@ impl candle::CustomOp1 for Im2Col {
let dst_idx = dst_idx + h_k_idx * w_k;
for w_k_idx in 0..w_k {
let src_w = w_idx * stride + w_k_idx * dilation;
if padding != 0 && (src_w < padding || src_w >= h + padding) {
if padding != 0 && (src_w < padding || src_w >= w + padding) {
continue;
}
let src_w = src_w - padding;