Skip to content

Commit

Permalink
See if type piracy removal fixes docs
Browse files Browse the repository at this point in the history
  • Loading branch information
ChrisRackauckas authored Mar 30, 2024
1 parent d94d935 commit c2b0b4b
Showing 1 changed file with 0 additions and 10 deletions.
10 changes: 0 additions & 10 deletions src/DiffEqFlux.jl
Original file line number Diff line number Diff line change
Expand Up @@ -19,16 +19,6 @@ import Lux.Experimental: StatefulLuxLayer

@reexport using ADTypes, Lux

# FIXME: Type Piracy
function CRC.rrule(::Type{Tridiagonal}, dl, d, du)
y = Tridiagonal(dl, d, du)
@views function ∇Tridiagonal(∂y)
return (NoTangent(), diag(∂y[2:end, 1:(end - 1)]), diag(∂y),
diag(∂y[1:(end - 1), 2:end]))
end
return y, ∇Tridiagonal
end

include("ffjord.jl")
include("neural_de.jl")
include("spline_layer.jl")
Expand Down

0 comments on commit c2b0b4b

Please sign in to comment.