Bijectors, reverse mode hangs

[mhauru]

MWE:

module MWE
import Bijectors, Enzyme
b = Bijectors.CorrBijector()
binv = Bijectors.inverse(b)
f(x) = sum(b(binv(x)))
x = randn((0, 0))
Enzyme.gradient(Enzyme.Reverse, f, x)
end

Output:

WARNING: replacing module MWE.
 ** On entry to DTRMM  parameter number  9 had an illegal value
 ** On entry to DLASCL parameter number  9 had an illegal value
 ** On entry to DTRSM  parameter number  9 had an illegal value
 ** On entry to DTRSM  parameter number  9 had an illegal value

It seems to just hang after that. I ran out of patience after 5 minutes.

Enzyme current main branch.

[wsmoses]

Okay this one requires minimization and isn't resolved by current dev branches

[wsmoses]

@penelopeysm and also would you be able to help reduce this one too?

[mhauru]

Giving this a shot now

[mhauru]

It's an issue with Cholesky on a zero-dim matrix:

module MWE
import Enzyme, LinearAlgebra

function cholesky!(A)
    C, info = LinearAlgebra._chol!(A, LinearAlgebra.UpperTriangular)
    return C
end

f(x) = sum(cholesky!(copy(x)))
x = zeros((0, 0))
Enzyme.gradient(Enzyme.Reverse, f, x)
end

[wsmoses]

fixed on main