Skip to content

Commit

Permalink
Merge pull request #914 from SciML/ChrisRackauckas-patch-2
Browse files Browse the repository at this point in the history
See if type piracy removal fixes docs
  • Loading branch information
ChrisRackauckas authored Mar 30, 2024
2 parents 7ae82fa + 96b9002 commit fd806fe
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 17 deletions.
14 changes: 7 additions & 7 deletions docs/Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
[compat]
CSV = "0.10"
CUDA = "5"
ComponentArrays = "0.13, 0.14, 0.15"
ComponentArrays = "0.15"
DataDeps = "0.7"
DataFrames = "1"
DiffEqFlux = "3"
Expand All @@ -49,27 +49,27 @@ Documenter = "1"
Flux = "0.14"
ForwardDiff = "0.10"
IterTools = "1"
LinearAlgebra = "<0.0.1, 1"
LinearAlgebra = "1"
Lux = "0.5.5"
LuxCUDA = "0.3"
MLDataUtils = "0.5"
MLDatasets = "0.7"
MLUtils = "0.4"
NNlib = "0.9"
Optimisers = "0.2, 0.3"
Optimisers = "0.3"
Optimization = "3.9"
OptimizationOptimJL = "0.2"
OptimizationOptimisers = "0.2"
OptimizationPolyalgorithms = "0.2"
OrdinaryDiffEq = "6.31"
Plots = "1.36"
Printf = "<0.0.1, 1"
Random = "<0.0.1, 1"
Printf = "1"
Random = "1"
ReverseDiff = "1.14"
SciMLBase = "1.72, 2"
SciMLBase = "2"
SciMLSensitivity = "7.11"
StableRNGs = "1"
Statistics = "1"
StochasticDiffEq = "6.56"
Test = "<0.0.1, 1"
Test = "1"
Zygote = "0.6.62"
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 fd806fe

Please sign in to comment.