Merge pull request #932 from SciML/ChrisRackauckas-patch-1

Remove recompile_invalidations
SciML · Jun 27, 2024 · efeb30e · efeb30e
2 parents 2691d59 + 2d88272
commit efeb30e
Showing 1 changed file with 24 additions and 29 deletions.
diff --git a/src/DiffEqFlux.jl b/src/DiffEqFlux.jl
@@ -1,34 +1,29 @@
 module DiffEqFlux
 
-using PrecompileTools: @recompile_invalidations
-
-@recompile_invalidations begin
- using ADTypes: ADTypes, AutoForwardDiff, AutoZygote
- using ChainRulesCore: ChainRulesCore
- using ConcreteStructs: @concrete
- using Distributions: Distributions, ContinuousMultivariateDistribution, Distribution,
- logpdf
- using DistributionsAD: DistributionsAD
- using ForwardDiff: ForwardDiff
- using Functors: Functors, fmap
- using LinearAlgebra: LinearAlgebra, Diagonal, det, tr, mul!
- using Lux: Lux, Chain, Dense, StatefulLuxLayer, FromFluxAdaptor, ⊠
- using LuxCore: LuxCore, AbstractExplicitLayer, AbstractExplicitContainerLayer
- using Random: Random, AbstractRNG, randn!
- using Reexport: @reexport
- using SciMLBase: SciMLBase, DAEProblem, DDEFunction, DDEProblem, EnsembleProblem,
- ODEFunction, ODEProblem, ODESolution, SDEFunction, SDEProblem, remake,
- solve
- using SciMLSensitivity: SciMLSensitivity, AdjointLSS, BacksolveAdjoint, EnzymeVJP,
- ForwardDiffOverAdjoint, ForwardDiffSensitivity, ForwardLSS,
- ForwardSensitivity, GaussAdjoint, InterpolatingAdjoint, NILSAS,
- NILSS, QuadratureAdjoint, ReverseDiffAdjoint, ReverseDiffVJP,
- SteadyStateAdjoint, TrackerAdjoint, TrackerVJP, ZygoteAdjoint,
- ZygoteVJP
- using Setfield: @set
- using Tracker: Tracker
- using Zygote: Zygote
-end
+using ADTypes: ADTypes, AutoForwardDiff, AutoZygote
+using ChainRulesCore: ChainRulesCore
+using ConcreteStructs: @concrete
+using Distributions: Distributions, ContinuousMultivariateDistribution, Distribution, logpdf
+using DistributionsAD: DistributionsAD
+using ForwardDiff: ForwardDiff
+using Functors: Functors, fmap
+using LinearAlgebra: LinearAlgebra, Diagonal, det, tr, mul!
+using Lux: Lux, Chain, Dense, StatefulLuxLayer, FromFluxAdaptor, ⊠
+using LuxCore: LuxCore, AbstractExplicitLayer, AbstractExplicitContainerLayer
+using Random: Random, AbstractRNG, randn!
+using Reexport: @reexport
+using SciMLBase: SciMLBase, DAEProblem, DDEFunction, DDEProblem, EnsembleProblem,
+ ODEFunction, ODEProblem, ODESolution, SDEFunction, SDEProblem, remake,
+ solve
+using SciMLSensitivity: SciMLSensitivity, AdjointLSS, BacksolveAdjoint, EnzymeVJP,
+ ForwardDiffOverAdjoint, ForwardDiffSensitivity, ForwardLSS,
+ ForwardSensitivity, GaussAdjoint, InterpolatingAdjoint, NILSAS,
+ NILSS, QuadratureAdjoint, ReverseDiffAdjoint, ReverseDiffVJP,
+ SteadyStateAdjoint, TrackerAdjoint, TrackerVJP, ZygoteAdjoint,
+ ZygoteVJP
+using Setfield: @set
+using Tracker: Tracker
+using Zygote: Zygote
 
 const CRC = ChainRulesCore