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

Bugfix.

Browse Source
This commit is contained in:
laurent
2023-09-09 19:11:59 +01:00
parent b283b3e181
commit c22f23e568
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/cpu_backend.rs
View File

@ -1220,7 +1220,7 @@ impl<'a> Map2 for ConvTranspose2D<'a> {
for k_x in 0..p.k_w {
(0..c_out).into_par_iter().for_each(|dst_c_idx| {
let (group_idx, dst_c_idx_in_group) =
(c_out / p.c_out_per_group, c_out % p.c_out_per_group);
(dst_c_idx / p.c_out_per_group, dst_c_idx % p.c_out_per_group);
let k_cont = (0..c_in_per_group)
.map(|c_in_idx| {
let c_in_idx = group_idx * c_in_per_group + c_in_idx;