Pin JET version

Author: ErikQQY

Project.toml

@@ -46,7 +46,7 @@ FastClosures = "0.3.2"
 ForwardDiff = "0.10.38"
 Hwloc = "3.3"
 InteractiveUtils = "<0.0.1, 1"
-JET = "0.9"
+JET = "0.9.19"
 LinearAlgebra = "1.10"
 LinearSolve = "3.9"
 NonlinearSolveFirstOrder = "1.3"

ext/BoundaryValueDiffEqODEInterfaceExt.jl

@@ -61,13 +61,17 @@ function SciMLBase.__solve(prob::BVProblem, alg::BVPM2; dt = 0.0, reltol = 1e-3,
         @closure (t, u, du) -> du .= vec(prob.f(reshape(u, u0_size), prob.p, t))
     end
     bvp2m_bc = if SciMLBase.isinplace(prob)
-        @closure (ya, yb, bca, bcb) -> begin
+        @closure (ya,
+            yb,
+            bca,
+            bcb) -> begin
             prob.f.bc[1](reshape(bca, left_bc_size), reshape(ya, u0_size), prob.p)
             prob.f.bc[2](reshape(bcb, right_bc_size), reshape(yb, u0_size), prob.p)
             return nothing
         end
     else
-        @closure (ya, yb, bca, bcb) -> begin
+        @closure (
+            ya, yb, bca, bcb) -> begin
             bca .= vec(prob.f.bc[1](reshape(ya, u0_size), prob.p))
             bcb .= vec(prob.f.bc[2](reshape(yb, u0_size), prob.p))
             return nothing
@@ -140,15 +144,18 @@ function SciMLBase.__solve(prob::BVProblem, alg::BVPSOL; maxiters = 1000,
     end
 
     bvpsol_bc = if SciMLBase.isinplace(prob)
-        @closure (ya, yb, r) -> begin
+        @closure (ya, yb,
+            r) -> begin
             left_bc = reshape(@view(r[1:no_left_bc]), left_bc_size)
             right_bc = reshape(@view(r[(no_left_bc + 1):end]), right_bc_size)
             prob.f.bc[1](left_bc, reshape(ya, u0_size), prob.p)
             prob.f.bc[2](right_bc, reshape(yb, u0_size), prob.p)
             return nothing
         end
     else
-        @closure (ya, yb, r) -> begin
+        @closure (ya,
+            yb,
+            r) -> begin
             r[1:no_left_bc] .= vec(prob.f.bc[1](reshape(ya, u0_size), prob.p))
             r[(no_left_bc + 1):end] .= vec(prob.f.bc[2](reshape(yb, u0_size), prob.p))
             return nothing

lib/BoundaryValueDiffEqAscher/Project.toml

@@ -32,7 +32,7 @@ FastClosures = "0.3.2"
 ForwardDiff = "0.10.38"
 Hwloc = "3"
 InteractiveUtils = "<0.0.1, 1"
-JET = "0.9"
+JET = "0.9.19"
 LinearAlgebra = "1.10"
 PreallocationTools = "0.4.24"
 Random = "1.10"

lib/BoundaryValueDiffEqAscher/src/adaptivity.jl

@@ -188,15 +188,13 @@ function error_estimate!(cache::AscherCache)
         # in valstr in case they prove to be needed later for an error estimate.
         x = mesh[i] + (mesh_dt[i]) * 2.0 / 3.0
         @views approx(cache, x, valstr[i][3])
-        error[i] .= wgterr .*
-                    abs.(valstr[i][3] .-
+        error[i] .= wgterr .* abs.(valstr[i][3] .-
                          (isodd(i) ? valstr[Int((i + 1) / 2)][2] : valstr[Int(i / 2)][4]))
 
         x = mesh[i] + (mesh_dt[i]) / 3.0
         @views approx(cache, x, valstr[i][2])
         error[i] .= error[i] .+
-                    wgterr .*
-                    abs.(valstr[i][2] .-
+                    wgterr .* abs.(valstr[i][2] .-
                          (isodd(i) ? valstr[Int((i + 1) / 2)][1] : valstr[Int(i / 2)][3]))
     end
     return maximum(reduce(hcat, error), dims = 2)

lib/BoundaryValueDiffEqAscher/src/algorithms.jl

@@ -18,6 +18,7 @@ for stage in (1, 2, 3, 4, 5, 6, 7)
           - `zeta`: side condition points, should always be provided.
 
         !!! note
+
             For type-stability, the chunksizes for ForwardDiff ADTypes in
             `BVPJacobianAlgorithm` must be provided.
 

lib/BoundaryValueDiffEqAscher/src/ascher.jl

@@ -100,7 +100,8 @@ function SciMLBase.__init(prob::BVProblem, alg::AbstractAscher; dt = 0.0,
 
     iip = isinplace(prob)
 
-    f, bc = if prob.u0 isa AbstractVector
+    f,
+    bc = if prob.u0 isa AbstractVector
         prob.f, prob.f.bc
     elseif iip
         vecf! = @closure (du, u, p, t) -> __vec_f!(du, u, p, t, prob.f, size(u0))
@@ -160,8 +161,9 @@ function SciMLBase.solve!(cache::AscherCache{iip, T}) where {iip, T}
 
     if adaptive
         while SciMLBase.successful_retcode(info) && norm(error_norm) > abstol
-            z, y, info, error_norm = __perform_ascher_iteration(
-                cache, abstol, adaptive; kwargs...)
+            z, y,
+            info,
+            error_norm = __perform_ascher_iteration(cache, abstol, adaptive; kwargs...)
         end
     end
     u = [vcat(zᵢ, yᵢ) for (zᵢ, yᵢ) in zip(z, y)]
@@ -338,10 +340,11 @@ function __construct_nlproblem(cache::AscherCache{iip, T}) where {iip, T}
     end
 
     jac = if iip
-        @closure (J, u, p) -> __ascher_mpoint_jacobian!(
-            J, u, diffmode, jac_cache, loss, lz, cache.p)
+        @closure (J, u,
+            p) -> __ascher_mpoint_jacobian!(J, u, diffmode, jac_cache, loss, lz, cache.p)
     else
-        @closure (u, p) -> __ascher_mpoint_jacobian(
+        @closure (u,
+            p) -> __ascher_mpoint_jacobian(
             jac_prototype, u, diffmode, jac_cache, loss, cache.p)
     end
 

lib/BoundaryValueDiffEqAscher/src/collocation.jl

@@ -1,5 +1,6 @@
 function Φ!(cache::AscherCache{iip, T}, z, res, pt::StandardBVProblem) where {iip, T}
-    (; f, mesh, mesh_dt, ncomp, ny, bc, k, p, zeta, residual, zval, yval, gval, delz, dmz, deldmz, g, w, v, ipvtg, ipvtw, TU) = cache
+    (; f, mesh, mesh_dt, ncomp, ny, bc, k, p, zeta, residual, zval,
+        yval, gval, delz, dmz, deldmz, g, w, v, ipvtg, ipvtw, TU) = cache
     (; acol, rho) = TU
     ncy = ncomp + ny
     n = length(mesh) - 1
@@ -156,7 +157,8 @@ function Φ!(cache::AscherCache{iip, T}, z, res, pt::StandardBVProblem) where {i
 end
 
 function Φ!(cache::AscherCache{iip, T}, z, res, pt::TwoPointBVProblem) where {iip, T}
-    (; f, mesh, mesh_dt, ncomp, ny, bc, k, p, zeta, bcresid_prototype, residual, zval, yval, gval, delz, dmz, deldmz, g, w, v, dmzo, ipvtg, ipvtw, TU) = cache
+    (; f, mesh, mesh_dt, ncomp, ny, bc, k, p, zeta, bcresid_prototype, residual,
+        zval, yval, gval, delz, dmz, deldmz, g, w, v, dmzo, ipvtg, ipvtw, TU) = cache
     (; acol, rho) = TU
     ncy = ncomp + ny
     n = length(mesh) - 1
@@ -319,7 +321,8 @@ end
 @inline __get_value(z) = isa(z, ForwardDiff.Dual) ? z.value : z
 
 function Φ(cache::AscherCache{iip, T}, z, pt::StandardBVProblem) where {iip, T}
-    (; f, mesh, mesh_dt, ncomp, ny, bc, k, p, zeta, residual, zval, yval, gval, delz, dmz, deldmz, g, w, v, dmzo, ipvtg, ipvtw, TU) = cache
+    (; f, mesh, mesh_dt, ncomp, ny, bc, k, p, zeta, residual, zval, yval,
+        gval, delz, dmz, deldmz, g, w, v, dmzo, ipvtg, ipvtw, TU) = cache
     (; acol, rho) = TU
     ncy = ncomp + ny
     n = length(mesh) - 1
@@ -476,7 +479,8 @@ function Φ(cache::AscherCache{iip, T}, z, pt::StandardBVProblem) where {iip, T}
 end
 
 function Φ(cache::AscherCache{iip, T}, z, pt::TwoPointBVProblem) where {iip, T}
-    (; f, mesh, mesh_dt, ncomp, ny, bc, k, p, zeta, residual, zval, yval, gval, delz, dmz, deldmz, g, w, v, dmzo, ipvtg, ipvtw, TU) = cache
+    (; f, mesh, mesh_dt, ncomp, ny, bc, k, p, zeta, residual, zval, yval,
+        gval, delz, dmz, deldmz, g, w, v, dmzo, ipvtg, ipvtw, TU) = cache
     (; acol, rho) = TU
     ncy = ncomp + ny
     n = length(mesh) - 1

lib/BoundaryValueDiffEqCore/Project.toml

@@ -32,7 +32,7 @@ ConcreteStructs = "0.2.3"
 DiffEqBase = "6.167"
 ForwardDiff = "0.10.38"
 InteractiveUtils = "<0.0.1, 1"
-JET = "0.9"
+JET = "0.9.19"
 LineSearch = "0.1.4"
 LinearAlgebra = "1.10"
 Logging = "1.10"

lib/BoundaryValueDiffEqCore/src/types.jl

@@ -6,7 +6,8 @@
 end
 
 @inline __materialize_jacobian_algorithm(_, alg::BVPJacobianAlgorithm) = alg
-@inline __materialize_jacobian_algorithm(_, alg::ADTypes.AbstractADType) = BVPJacobianAlgorithm(alg)
+@inline __materialize_jacobian_algorithm(
+    _, alg::ADTypes.AbstractADType) = BVPJacobianAlgorithm(alg)
 @inline __materialize_jacobian_algorithm(::Nothing, ::Nothing) = BVPJacobianAlgorithm()
 @inline function __materialize_jacobian_algorithm(nlsolve::N, ::Nothing) where {N}
     ad = hasfield(N, :jacobian_ad) ? nlsolve.jacobian_ad : missing

lib/BoundaryValueDiffEqFIRK/Project.toml

@@ -41,7 +41,7 @@ FastClosures = "0.3.2"
 ForwardDiff = "0.10.38"
 Hwloc = "3"
 InteractiveUtils = "<0.0.1, 1"
-JET = "0.9"
+JET = "0.9.19"
 LinearAlgebra = "1.10"
 LinearSolve = "2.36.2, 3"
 OrdinaryDiffEq = "6.93"