Skip to content

Commit ae4866e

Browse files
less unthunking
Co-authored-by: Michael Abbott <[email protected]>
1 parent 7d2987e commit ae4866e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/conv.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -331,11 +331,11 @@ end
331331

332332
function rrule(::typeof(∇conv_filter), x, dy, cdims; kw...)
333333
function ∇conv_filter_pullback(Δ)
334-
Δ = colmajor(Δ)
334+
Δ1 = colmajor(unthunk(Δ))
335335
return (
336336
NoTangent(),
337-
@thunk(∇conv_data(dy, unthunk(Δ), cdims, kw...)),
338-
@thunk(conv(x, unthunk(Δ), cdims, kw...)),
337+
@thunk(∇conv_data(dy, Δ1, cdims, kw...)),
338+
@thunk(conv(x, Δ1, cdims, kw...)),
339339
NoTangent(),
340340
)
341341
end

0 commit comments

Comments
 (0)