From c22f23e568c93d98df36b6a8a7f7481b4609f478 Mon Sep 17 00:00:00 2001 From: laurent Date: Sat, 9 Sep 2023 19:11:59 +0100 Subject: [PATCH] Bugfix. --- candle-core/src/cpu_backend.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/candle-core/src/cpu_backend.rs b/candle-core/src/cpu_backend.rs index 11250e82..c0f99363 100644 --- a/candle-core/src/cpu_backend.rs +++ b/candle-core/src/cpu_backend.rs @@ -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;