diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-02-19 09:04:49 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-19 09:04:49 +0100 |
| commit | fc67d878bb4a25cbeba361d0a31290f14beb9344 (patch) | |
| tree | c125a09771f8be9244adaa6c1c35d443a595fea7 /candle-core/src/cpu_backend.rs | |
| parent | 3ba37443e5e2bd8a31b84ba5335f3805581a631e (diff) | |
| download | candle-fc67d878bb4a25cbeba361d0a31290f14beb9344.tar.gz candle-fc67d878bb4a25cbeba361d0a31290f14beb9344.tar.bz2 candle-fc67d878bb4a25cbeba361d0a31290f14beb9344.zip | |
Bugfix for conv-transpose1d (#1734)
* Add a currently broken test.
* Bugfix + fix test.
Diffstat (limited to 'candle-core/src/cpu_backend.rs')
| -rw-r--r-- | candle-core/src/cpu_backend.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-core/src/cpu_backend.rs b/candle-core/src/cpu_backend.rs index f912c1b2..05e8c979 100644 --- a/candle-core/src/cpu_backend.rs +++ b/candle-core/src/cpu_backend.rs @@ -1263,6 +1263,7 @@ impl<'a> Map2 for ConvTranspose1D<'a> { fn f<T: WithDType>(&self, inp: &[T], inp_l: &Layout, k: &[T], k_l: &Layout) -> Result<Vec<T>> { let p = self.0; let inp = &inp[inp_l.start_offset()..]; + let k = &k[k_l.start_offset()..]; let (inp_s0, inp_s1, inp_s2) = crate::shape::dims3(inp_l.stride())?; let (k_s0, k_s1, k_s2) = crate::shape::dims3(k_l.stride())?; let l_out = p.l_out(); |