Skip to content

Commit 33e4959

Browse files
committed
fix: unthunk order
1 parent 81ec729 commit 33e4959

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

lib/LuxLib/src/impl/conv.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ CRC.@opt_out rrule(
196196

197197
function ∇fused_conv(Δ′, weight, x, bias, cdims::ConvDims, z, tmp, 𝒫w, 𝒫x, 𝒫b, act)
198198
old_threads = maybe_reduce_BLAS_threads(weight)
199-
Δ = CRC.unthunk(NNlib.colmajor(Δ′))
199+
Δ = NNlib.colmajor(CRC.unthunk(Δ′))
200200
∂y = ∇activation(Δ, z, act, tmp)
201201
∂w, ∂x, ∂b = ∇conv_bias(∂y, weight, x, bias, cdims)
202202
reset_BLAS_threads(old_threads)

0 commit comments

Comments
 (0)