Bijectors: Unknown concrete type in arraycopy_common. tt: {[]:Pointer}

[mhauru]

MWE:

julia> module MWE
       import Bijectors, Enzyme, Distributions, LinearAlgebra

       function f(θ)
           layer = Bijectors.PlanarLayer(θ[1:2], θ[3:4], θ[5:5])
           flow = Bijectors.transformed(Distributions.MvNormal(zeros(2), LinearAlgebra.I), Bijectors.inverse(layer))
           x = reshape(θ[6:end], 2, :)
           return sum(Distributions.logpdf(flow.dist, x) - Bijectors.logabsdetjac(flow.transform, x))
       end

       Enzyme.gradient(Enzyme.Reverse, f, randn(11))
       end

WARNING: replacing module MWE.
ERROR: Enzyme execution failed.
Enzyme: Unknown concrete type in arraycopy_common. tt: {[]:Pointer}   %42 = call noalias nonnull "enzyme_type"="{[-1]:Pointer}" {} addrspace(10)* @ijl_array_copy({} addrspace(10)* nonnull %30) #99, !dbg !260 (false, nothing, nothing)

Stacktrace:
  [1] copy
    @ ./array.jl:411 [inlined]
  [2] unaliascopy
    @ ./abstractarray.jl:1497 [inlined]
  [3] unalias
    @ ./abstractarray.jl:1481 [inlined]
  [4] broadcast_unalias
    @ ./broadcast.jl:977 [inlined]
  [5] preprocess
    @ ./broadcast.jl:984 [inlined]
  [6] preprocess_args
    @ ./broadcast.jl:986 [inlined]
  [7] preprocess
    @ ./broadcast.jl:983 [inlined]
  [8] preprocess_args
    @ ./broadcast.jl:987 [inlined]
  [9] preprocess
    @ ./broadcast.jl:983 [inlined]
 [10] preprocess_args
    @ ./broadcast.jl:987 [inlined]
 [11] preprocess
    @ ./broadcast.jl:983 [inlined]
 [12] preprocess_args
    @ ./broadcast.jl:987 [inlined]
 [13] preprocess_args
    @ ./broadcast.jl:986 [inlined]
 [14] preprocess
    @ ./broadcast.jl:983 [inlined]
 [15] preprocess_args
    @ ./broadcast.jl:987 [inlined]
 [16] preprocess
    @ ./broadcast.jl:983 [inlined]
 [17] copyto!
    @ ./broadcast.jl:1000 [inlined]
 [18] copyto!
    @ ./broadcast.jl:956 [inlined]
 [19] copy
    @ ./broadcast.jl:928 [inlined]
 [20] materialize
    @ ./broadcast.jl:903 [inlined]
 [21] with_logabsdet_jacobian
    @ ~/projects/Bijectors.jl/src/bijectors/planar_layer.jl:107
 [22] logabsdetjac
    @ ~/projects/Bijectors.jl/src/bijectors/planar_layer.jl:183 [inlined]
 [23] logabsdetjac
    @ ~/projects/Bijectors.jl/src/interface.jl:211 [inlined]
 [24] f
    @ ./REPL[5]:8 [inlined]
 [25] diffejulia_f_30158wrap
    @ ./REPL[5]:0
 [26] macro expansion
    @ ~/.julia/packages/Enzyme/BRtTP/src/compiler.jl:8137 [inlined]
 [27] enzyme_call
    @ ~/.julia/packages/Enzyme/BRtTP/src/compiler.jl:7703 [inlined]
 [28] CombinedAdjointThunk
    @ ~/.julia/packages/Enzyme/BRtTP/src/compiler.jl:7476 [inlined]
 [29] autodiff
    @ ~/.julia/packages/Enzyme/BRtTP/src/Enzyme.jl:491 [inlined]
 [30] autodiff
    @ ~/.julia/packages/Enzyme/BRtTP/src/Enzyme.jl:512 [inlined]
 [31] macro expansion
    @ ~/.julia/packages/Enzyme/BRtTP/src/Enzyme.jl:1719 [inlined]
 [32] gradient(::ReverseMode{false, false, FFIABI, false, false}, ::typeof(Main.MWE.f), ::Vector{Float64})
    @ Enzyme ~/.julia/packages/Enzyme/BRtTP/src/Enzyme.jl:1660
 [33] top-level scope
    @ REPL[5]:11

On Enzyme v0.13.12. This is a regression post-v.0.12.22, when it used to work.

[wsmoses]

Also probably needs minimization