From c2b0b4be3677eee10fb372492c620c0a0a9db936 Mon Sep 17 00:00:00 2001 From: Christopher Rackauckas Date: Sat, 30 Mar 2024 07:43:01 -0400 Subject: [PATCH] See if type piracy removal fixes docs --- src/DiffEqFlux.jl | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/src/DiffEqFlux.jl b/src/DiffEqFlux.jl index 06e9c59b0..cd6c451f5 100644 --- a/src/DiffEqFlux.jl +++ b/src/DiffEqFlux.jl @@ -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")