SciML / Integrals.jl

A common interface for quadrature and numerical integration for the SciML scientific machine learning organization
https://docs.sciml.ai/Integrals/stable/
MIT License
226 stars 30 forks source link

HCubature does not work with ODESolution produced by some ODE solvers #221

Open homocomputeris opened 9 months ago

homocomputeris commented 9 months ago

Describe the bug 🐞

HCubature breaks down when passed an ODESolution obtained wih Trapezoid()

Expected behavior

Calculate the cubature.

Minimal Reproducible Example 👇

Without MRE, we would only be able to help you to a limited extent, and attention to the issue would be limited. to know more about MRE refer to wikipedia and stackoverflow.

module Mwe

using DifferentialEquations
using Integrals
using Distributions
using LinearAlgebra
using ForwardDiff

# just to speed things up a bit
const TOL = 1e-6

# some parameters; exact values do not matter
tspan = (0, 2*pi + 0.0)
p = [1.0, 1.0]
t = 22.0 / 7.0
# limit the domain of interest
rectangle = (-5, 5)

function f(u, p, t)
    du = [u[2], -u[1], 0.0]
    return du
end

# function f!(du, u, p, t)
#     du[1] = u[2]
#     du[2] = u[1]
#     du[3] = 0.0
#     return nothing
# end

# function ff!(du, u, p, t)
#     du .= (u[2], u[1], 0.0)
#     return du # (or return nothing)
# end

# scalar function with vector argument
function g0(x, p, t)
    return pdf(truncated(Normal(1, 1), rectangle...), x[1]) *
           pdf(truncated(Normal(1, 1), rectangle...), x[2]) *
           pdf(truncated(Normal(1, 1), rectangle...), x[3])
end

function g1(x, p, t)
    if t > tspan[end]
        DomainError(t)
    end
    # find the initial condition for a given point
    function tmp(y)
        prob = ODEProblem(f, y, reverse(tspan), p)
        sol = solve(prob, Trapezoid(), dt = pi / 10)
        return sol(tspan[end] - t)
    end
    # use the found IC in g0
    return g0(tmp(x), p, 0.0) * 1
end

# Integrate everywhere
function x1(jpdf, x::Number, p, t)
    prob = IntegralProblem((tail, p) -> jpdf([x; tail], p, t),
        [-5, -Inf],
        [5, Inf], p)
    sol = solve(prob, HCubatureJL();
        reltol = TOL, abstol = TOL)
    return sol.u
end

# Truncate in a rectangle
function x1_broken(jpdf, x::Number, p, t)
    prob = IntegralProblem((tail, p) -> jpdf([x; tail], p, t),
        [-5, -5],
        [5, 5], p)
    sol = solve(prob, HCubatureJL();
        reltol = TOL, abstol = TOL)
    return sol.u
end

y = [1, 3, 1]
@show g0(y, p, 0)
@show g1(y, p, 0)
@show g1(y, p, 4)

# this works
@show x1(g1, 3.0, p, t)

# this doesn't
@show x1_broken(g1, 3.0, p, t)

end

Error & Stacktrace ⚠️

g0(y, p, 0) = 0.008593745726254109
g1(y, p, 0) = 0.008593745726254109
g1(y, p, 4) = 4.95625034962964e-5
ERROR: MethodError: Cannot `convert` an object of type LinearAlgebra.LU{Float64, Matrix{Float64}, Vector{Int64}} to an object of type StaticArrays.LU{LinearAlgebra.LowerTriangular{Float64, StaticArraysCore.SMatrix{3, 3, Float64, 9}}, LinearAlgebra.UpperTriangular{Float64, StaticArraysCore.SMatrix{3, 3, Float64, 9}}, StaticArraysCore.SVector{3, Int64}}

Closest candidates are:
  convert(::Type{T}, ::T) where T
   @ Base Base.jl:84
  (::Type{StaticArrays.LU{L, U, p}} where {L, U, p})(::Any, ::Any, ::Any)
   @ StaticArrays ~/.julia/packages/StaticArrays/eGKzB/src/lu.jl:3

Stacktrace:
  [1] setproperty!(x::OrdinaryDiffEq.NLNewtonConstantCache{…}, f::Symbol, v::LinearAlgebra.LU{…})
    @ Base ./Base.jl:40
  [2] update_W!(nlsolver::OrdinaryDiffEq.NLSolver{…}, integrator::OrdinaryDiffEq.ODEIntegrator{…}, cache::OrdinaryDiffEq.TrapezoidConstantCache{…}, dtgamma::Float64, repeat_step::Bool, newJW::Nothing)
    @ OrdinaryDiffEq ~/.julia/packages/OrdinaryDiffEq/2nLli/src/derivative_utils.jl:832
  [3] update_W!(nlsolver::OrdinaryDiffEq.NLSolver{…}, integrator::OrdinaryDiffEq.ODEIntegrator{…}, cache::OrdinaryDiffEq.TrapezoidConstantCache{…}, dtgamma::Float64, repeat_step::Bool)
    @ OrdinaryDiffEq ~/.julia/packages/OrdinaryDiffEq/2nLli/src/derivative_utils.jl:822
  [4] nlsolve!(nlsolver::OrdinaryDiffEq.NLSolver{…}, integrator::OrdinaryDiffEq.ODEIntegrator{…}, cache::OrdinaryDiffEq.TrapezoidConstantCache{…}, repeat_step::Bool)
    @ OrdinaryDiffEq ~/.julia/packages/OrdinaryDiffEq/2nLli/src/nlsolve/nlsolve.jl:27
  [5] perform_step!(integrator::OrdinaryDiffEq.ODEIntegrator{…}, cache::OrdinaryDiffEq.TrapezoidConstantCache{…}, repeat_step::Bool)
    @ OrdinaryDiffEq ~/.julia/packages/OrdinaryDiffEq/2nLli/src/perform_step/sdirk_perform_step.jl:230
  [6] perform_step!
    @ OrdinaryDiffEq ~/.julia/packages/OrdinaryDiffEq/2nLli/src/perform_step/sdirk_perform_step.jl:211 [inlined]
  [7] solve!(integrator::OrdinaryDiffEq.ODEIntegrator{…})
    @ OrdinaryDiffEq ~/.julia/packages/OrdinaryDiffEq/2nLli/src/solve.jl:537
  [8] __solve(::SciMLBase.ODEProblem{…}, ::OrdinaryDiffEq.Trapezoid{…}; kwargs::@Kwargs{…})
    @ OrdinaryDiffEq ~/.julia/packages/OrdinaryDiffEq/2nLli/src/solve.jl:6
  [9] __solve
    @ OrdinaryDiffEq ~/.julia/packages/OrdinaryDiffEq/2nLli/src/solve.jl:1 [inlined]
 [10] #solve_call#34
    @ DiffEqBase ~/.julia/packages/DiffEqBase/eLhx9/src/solve.jl:609 [inlined]
 [11] solve_call
    @ DiffEqBase ~/.julia/packages/DiffEqBase/eLhx9/src/solve.jl:567 [inlined]
 [12] #solve_up#42
    @ DiffEqBase ~/.julia/packages/DiffEqBase/eLhx9/src/solve.jl:1058 [inlined]
 [13] solve_up
    @ DiffEqBase ~/.julia/packages/DiffEqBase/eLhx9/src/solve.jl:1044 [inlined]
 [14] #solve#40
    @ DiffEqBase ~/.julia/packages/DiffEqBase/eLhx9/src/solve.jl:981 [inlined]
 [15] (::Main.Mwe.var"#tmp#1"{Vector{Float64}, Float64})(y::StaticArraysCore.MVector{3, Float64})
    @ Main.Mwe ~/Code/test/mwe.jl:50
 [16] g1(x::StaticArraysCore.MVector{3, Float64}, p::Vector{Float64}, t::Float64)
    @ Main.Mwe ~/Code/test/mwe.jl:54
 [17] (::Main.Mwe.var"#2#3"{…})(tail::StaticArraysCore.MVector{…}, p::Vector{…})
    @ Main.Mwe ~/Code/test/mwe.jl:59
 [18] IntegralFunction
    @ ~/.julia/packages/SciMLBase/XxTxA/src/scimlfunctions.jl:2358 [inlined]
 [19] substitute_f(t::StaticArraysCore.SVector{…}, p::Vector{…}, f::SciMLBase.IntegralFunction{…}, lb::Vector{…}, ub::Vector{…})
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/infinity_handling.jl:62
 [20] #19
    @ ~/.julia/packages/Integrals/d5Wr6/src/infinity_handling.jl:110 [inlined]
 [21] IntegralFunction
    @ ~/.julia/packages/SciMLBase/XxTxA/src/scimlfunctions.jl:2358 [inlined]
 [22] #46
    @ ~/.julia/packages/Integrals/d5Wr6/src/Integrals.jl:113 [inlined]
 [23] (::HCubature.GenzMalik{…})(f::Integrals.var"#46#48"{…}, a::StaticArraysCore.SVector{…}, b::StaticArraysCore.SVector{…}, norm::typeof(LinearAlgebra.norm))
    @ HCubature ~/.julia/packages/HCubature/QvyJW/src/genz-malik.jl:121
 [24] hcubature_(f::Integrals.var"#46#48"{…}, a::StaticArraysCore.SVector{…}, b::StaticArraysCore.SVector{…}, norm::typeof(LinearAlgebra.norm), rtol_::Float64, atol::Float64, maxevals::Int64, initdiv::Int64)
    @ HCubature ~/.julia/packages/HCubature/QvyJW/src/HCubature.jl:61
 [25] hcubature_(f::Integrals.var"#46#48"{…}, a::Vector{…}, b::Vector{…}, norm::Function, rtol::Float64, atol::Float64, maxevals::Int64, initdiv::Int64)
    @ HCubature ~/.julia/packages/HCubature/QvyJW/src/HCubature.jl:129
 [26] hcubature
    @ ~/.julia/packages/HCubature/QvyJW/src/HCubature.jl:178 [inlined]
 [27] __solvebp_call(prob::SciMLBase.IntegralProblem{…}, alg::Integrals.HCubatureJL{…}, sensealg::Integrals.ReCallVJP{…}, domain::Tuple{…}, p::Vector{…}; reltol::Float64, abstol::Float64, maxiters::Int64)
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/Integrals.jl:121
 [28] __solvebp_call
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/Integrals.jl:102 [inlined]
 [29] #__solvebp_call#4
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/common.jl:113 [inlined]
 [30] __solvebp_call
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/common.jl:112 [inlined]
 [31] __solvebp
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/Integrals.jl:65 [inlined]
 [32] solve!(cache::Integrals.IntegralCache{…})
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/common.jl:103
 [33] solve(prob::SciMLBase.IntegralProblem{…}, alg::Integrals.HCubatureJL{…}; kwargs::@Kwargs{…})
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/common.jl:99
 [34] solve
    @ ~/.julia/packages/Integrals/d5Wr6/src/common.jl:96 [inlined]
 [35] x1(jpdf::typeof(Main.Mwe.g1), x::Float64, p::Vector{Float64}, t::Float64)
    @ Main.Mwe ~/Code/test/mwe.jl:62
 [36] macro expansion
    @ show.jl:1181 [inlined]
 [37] top-level scope
    @ ~/Code/test/mwe.jl:83
Some type information was truncated. Use `show(err)` to see complete types.

Environment (please complete the following information):

  [7e558dbc] ArbNumerics v1.3.3
  [667455a9] Cubature v1.5.1
  [0c46a032] DifferentialEquations v7.12.0
  [31c24e10] Distributions v0.25.107
  [f6369f11] ForwardDiff v0.10.36
  [de52edbc] Integrals v4.1.0
  [91a5bcdd] Plots v1.40.0
  [0d4725de] Readables v0.3.3
 [47edcb42] ADTypes v0.2.6
⌅ [79e6a3ab] Adapt v3.7.2
  [7e558dbc] ArbNumerics v1.3.3
  [ec485272] ArnoldiMethod v0.2.0
  [4fba245c] ArrayInterface v7.7.0
  [4c555306] ArrayLayouts v1.5.2
  [aae01518] BandedMatrices v1.4.0
  [6e4b80f9] BenchmarkTools v1.4.0
  [d1d4a3ce] BitFlags v0.1.8
  [62783981] BitTwiddlingConvenienceFunctions v0.1.5
⌃ [764a87c0] BoundaryValueDiffEq v5.6.0
  [fa961155] CEnum v0.5.0
  [2a0fbf3d] CPUSummary v0.2.4
  [49dc2e85] Calculus v0.5.1
  [fb6a15b2] CloseOpenIntervals v0.1.12
  [523fee87] CodecBzip2 v0.8.1
  [944b1d66] CodecZlib v0.7.3
  [35d6a980] ColorSchemes v3.24.0
  [3da002f7] ColorTypes v0.11.4
  [c3611d14] ColorVectorSpace v0.10.0
  [5ae59095] Colors v0.12.10
  [861a8166] Combinatorics v1.0.2
  [38540f10] CommonSolve v0.2.4
  [bbf7d656] CommonSubexpressions v0.3.0
  [34da2185] Compat v4.12.0
  [2569d6c7] ConcreteStructs v0.2.3
  [f0e56b4a] ConcurrentUtilities v2.3.0
  [187b0558] ConstructionBase v1.5.4
  [d38c429a] Contour v0.6.2
  [adafc99b] CpuId v0.3.1
  [667455a9] Cubature v1.5.1
  [9a962f9c] DataAPI v1.16.0
  [864edb3b] DataStructures v0.18.16
  [e2d170a0] DataValueInterfaces v1.0.0
  [bcd4f6db] DelayDiffEq v5.46.0
  [8bb1440f] DelimitedFiles v1.9.1
  [2b5f629d] DiffEqBase v6.146.0
  [459566f4] DiffEqCallbacks v2.36.1
  [77a26b50] DiffEqNoiseProcess v5.20.0
  [163ba53b] DiffResults v1.1.0
  [b552c78f] DiffRules v1.15.1
  [0c46a032] DifferentialEquations v7.12.0
  [b4f34e82] Distances v0.10.11
  [31c24e10] Distributions v0.25.107
  [ffbed154] DocStringExtensions v0.9.3
  [fa6b7ba4] DualNumbers v0.6.8
  [4e289a0a] EnumX v1.0.4
  [f151be2c] EnzymeCore v0.6.5
  [460bff9d] ExceptionUnwrapping v0.1.10
  [d4d017d3] ExponentialUtilities v1.25.0
  [e2ba6199] ExprTools v0.1.10
  [c87230d0] FFMPEG v0.4.1
  [9d29842c] FastAlmostBandedMatrices v0.1.0
  [7034ab61] FastBroadcast v0.2.8
  [9aa1b823] FastClosures v0.3.2
  [29a986be] FastLapackInterface v2.0.0
  [1a297f60] FillArrays v1.9.3
  [6a86dc24] FiniteDiff v2.22.0
  [53c48c17] FixedPointNumbers v0.8.4
  [59287772] Formatting v0.4.2
  [f6369f11] ForwardDiff v0.10.36
  [069b7b12] FunctionWrappers v1.1.3
  [77dc65aa] FunctionWrappersWrappers v0.1.3
  [d9f16b24] Functors v0.4.5
⌃ [46192b85] GPUArraysCore v0.1.5
  [28b8d3ca] GR v0.73.1
  [14197337] GenericLinearAlgebra v0.3.11
  [c145ed77] GenericSchur v0.5.3
  [86223c79] Graphs v1.9.0
  [42e2da0e] Grisu v1.0.2
  [19dc6840] HCubature v1.5.1
  [cd3eb016] HTTP v1.10.1
  [3e5b6fbb] HostCPUFeatures v0.1.16
  [34004b35] HypergeometricFunctions v0.3.23
  [615f187c] IfElse v0.1.1
  [d25df0c9] Inflate v0.1.4
  [de52edbc] Integrals v4.1.0
  [92d709cd] IrrationalConstants v0.2.2
  [82899510] IteratorInterfaceExtensions v1.0.0
  [1019f520] JLFzf v0.1.7
  [692b3bcd] JLLWrappers v1.5.0
  [682c06a0] JSON v0.21.4
  [ccbc3e58] JumpProcesses v9.10.1
  [ef3ab10e] KLU v0.4.1
  [ba0b0d4f] Krylov v0.9.5
  [b964fa9f] LaTeXStrings v1.3.1
  [23fbe1c1] Latexify v0.16.1
  [a5e1c1ea] LatinHypercubeSampling v1.9.0
  [73f95e8e] LatticeRules v0.0.1
  [10f19ff3] LayoutPointers v0.1.15
  [50d2b5c4] Lazy v0.15.1
  [5078a376] LazyArrays v1.8.3
  [2d8b4e74] LevyArea v1.0.0
  [d3d80556] LineSearches v7.2.0
  [7ed4a6bd] LinearSolve v2.22.1
  [2ab3a3ac] LogExpFunctions v0.3.26
  [e6f89c97] LoggingExtras v1.0.3
  [bdcacae8] LoopVectorization v0.12.166
  [1914dd2f] MacroTools v0.5.13
  [d125e4d3] ManualMemory v0.1.8
  [b8f27783] MathOptInterface v1.25.1
  [a3b82374] MatrixFactorizations v2.1.0
  [bb5d69b7] MaybeInplace v0.1.1
  [739be429] MbedTLS v1.1.9
  [442fdcdd] Measures v0.3.2
  [e1d29d7a] Missings v1.1.0
  [4886b29c] MonteCarloIntegration v0.1.2
  [46d2c3a1] MuladdMacro v0.2.4
  [d8a4904e] MutableArithmetics v1.4.0
  [d41bc354] NLSolversBase v7.8.3
  [2774e3e8] NLsolve v4.5.1
  [77ba4419] NaNMath v1.0.2
  [8913a72c] NonlinearSolve v3.5.0
  [6fe1bfb0] OffsetArrays v1.13.0
  [4d8831e6] OpenSSL v1.4.1
  [429524aa] Optim v1.8.0
  [bac558e1] OrderedCollections v1.6.3
  [1dea7af3] OrdinaryDiffEq v6.69.0
  [90014a1f] PDMats v0.11.31
  [65ce6f38] PackageExtensionCompat v1.0.2
  [d96e819e] Parameters v0.12.3
  [69de0a69] Parsers v2.8.1
  [b98c9c47] Pipe v1.3.0
  [ccf2f8ad] PlotThemes v3.1.0
  [995b91a9] PlotUtils v1.4.0
  [91a5bcdd] Plots v1.40.0
  [e409e4f3] PoissonRandom v0.4.4
  [f517fe37] Polyester v0.7.9
  [1d0040c9] PolyesterWeave v0.2.1
  [85a6dd25] PositiveFactorizations v0.2.4
  [d236fae5] PreallocationTools v0.4.17
  [aea7be01] PrecompileTools v1.2.0
  [21216c6a] Preferences v1.4.1
  [1fd47b50] QuadGK v2.9.4
⌅ [8a4e6c94] QuasiMonteCarlo v0.2.9
  [74087812] Random123 v1.6.2
  [e6cf234a] RandomNumbers v1.5.3
  [0d4725de] Readables v0.3.3
  [3cdcf5f2] RecipesBase v1.3.4
  [01d81517] RecipesPipeline v0.6.12
  [731186ca] RecursiveArrayTools v3.6.2
  [f2c3362d] RecursiveFactorization v0.2.21
  [189a3867] Reexport v1.2.2
  [05181044] RelocatableFolders v1.0.1
  [ae029012] Requires v1.3.0
  [ae5879a3] ResettableStacks v1.1.1
  [79098fc4] Rmath v0.7.1
  [7e49a35a] RuntimeGeneratedFunctions v0.5.12
  [94e857df] SIMDTypes v0.1.0
  [476501e8] SLEEFPirates v0.6.42
  [0bca4576] SciMLBase v2.21.0
  [c0aeaf25] SciMLOperators v0.3.7
  [6c6a2e73] Scratch v1.2.1
  [efcf1570] Setfield v1.1.1
  [992d4aef] Showoff v1.0.3
  [777ac1f9] SimpleBufferStream v1.1.0
  [727e6d20] SimpleNonlinearSolve v1.3.1
  [699a6c99] SimpleTraits v0.9.4
  [ce78b400] SimpleUnPack v1.1.0
  [ed01d8cd] Sobol v1.5.0
  [a2af1166] SortingAlgorithms v1.2.1
  [47a9eef4] SparseDiffTools v2.16.0
  [e56a9233] Sparspak v0.3.9
  [276daf66] SpecialFunctions v2.3.1
  [860ef19b] StableRNGs v1.0.1
  [aedffcd0] Static v0.8.8
  [0d7ed370] StaticArrayInterface v1.5.0
  [90137ffa] StaticArrays v1.9.1
  [1e83bf80] StaticArraysCore v1.4.2
  [82ae8749] StatsAPI v1.7.0
  [2913bbd2] StatsBase v0.34.2
  [4c63d2b9] StatsFuns v1.3.0
  [9672c7b4] SteadyStateDiffEq v2.0.1
  [789caeaf] StochasticDiffEq v6.64.0
  [7792a7ef] StrideArraysCore v0.5.2
  [c3572dad] Sundials v4.23.1
  [2efcf032] SymbolicIndexingInterface v0.3.4
  [3783bdb8] TableTraits v1.0.1
  [bd369af6] Tables v1.11.1
  [62fd8b95] TensorCore v0.1.1
  [8290d209] ThreadingUtilities v0.5.2
  [a759f4b9] TimerOutputs v0.5.23
  [3bb67fe8] TranscodingStreams v0.10.2
  [d5829a12] TriangularSolve v0.1.20
  [410a4b4d] Tricks v0.1.8
  [781d530d] TruncatedStacktraces v1.4.0
  [5c2747f8] URIs v1.5.1
  [3a884ed6] UnPack v1.0.2
  [1cfade01] UnicodeFun v0.4.1
  [1986cc42] Unitful v1.19.0
  [45397f5d] UnitfulLatexify v1.6.3
  [41fe7b60] Unzip v0.2.0
  [3d5dd08c] VectorizationBase v0.21.65
  [19fa3120] VertexSafeGraphs v0.2.0
  [d9960996] Arb_jll v200.2300.0+0
  [6e34b625] Bzip2_jll v1.0.8+1
  [83423d85] Cairo_jll v1.16.1+1
  [7bc98958] Cubature_jll v1.0.5+0
  [2702e6a9] EpollShim_jll v0.0.20230411+0
  [2e619515] Expat_jll v2.5.0+0
  [b22a6f82] FFMPEG_jll v4.4.4+1
⌅ [e134572f] FLINT_jll v200.900.9+0
  [a3f928ae] Fontconfig_jll v2.13.93+0
  [d7e528f0] FreeType2_jll v2.13.1+0
  [559328eb] FriBidi_jll v1.0.10+0
  [0656b61e] GLFW_jll v3.3.9+0
  [d2c73de3] GR_jll v0.73.1+0
  [78b55507] Gettext_jll v0.21.0+0
  [7746bdde] Glib_jll v2.76.5+0
  [3b182d85] Graphite2_jll v1.3.14+0
  [2e76f6c2] HarfBuzz_jll v2.8.1+1
  [1d5cc7b8] IntelOpenMP_jll v2024.0.2+0
  [aacddb02] JpegTurbo_jll v3.0.1+0
  [c1c5ebd0] LAME_jll v3.100.1+0
  [88015f11] LERC_jll v3.0.0+1
  [1d63c593] LLVMOpenMP_jll v15.0.7+0
  [dd4b983a] LZO_jll v2.10.1+0
⌅ [e9f186c6] Libffi_jll v3.2.2+1
  [d4300ac3] Libgcrypt_jll v1.8.7+0
  [7e76a0d4] Libglvnd_jll v1.6.0+0
  [7add5ba3] Libgpg_error_jll v1.42.0+0
  [94ce4f54] Libiconv_jll v1.17.0+0
  [4b2f31a3] Libmount_jll v2.35.0+0
⌅ [89763e89] Libtiff_jll v4.5.1+1
  [38a345b3] Libuuid_jll v2.36.0+0
  [856f044c] MKL_jll v2024.0.0+0
  [e7412a2a] Ogg_jll v1.3.5+1
⌅ [656ef2d0] OpenBLAS32_jll v0.3.24+0
  [458c3c95] OpenSSL_jll v3.0.12+0
  [efe28fd5] OpenSpecFun_jll v0.5.5+0
  [91d4177d] Opus_jll v1.3.2+0
  [30392449] Pixman_jll v0.42.2+0
  [c0090381] Qt6Base_jll v6.5.3+1
  [f50d1b31] Rmath_jll v0.4.0+0
⌅ [fb77eaff] Sundials_jll v5.2.2+0
  [a44049a8] Vulkan_Loader_jll v1.3.243+0
  [a2964d1f] Wayland_jll v1.21.0+1
  [2381bf8a] Wayland_protocols_jll v1.31.0+0
  [02c8fc9c] XML2_jll v2.12.2+0
  [aed1982a] XSLT_jll v1.1.34+0
  [ffd25f8a] XZ_jll v5.4.5+0
  [f67eecfb] Xorg_libICE_jll v1.0.10+1
  [c834827a] Xorg_libSM_jll v1.2.3+0
  [4f6342f7] Xorg_libX11_jll v1.8.6+0
  [0c0b7dd1] Xorg_libXau_jll v1.0.11+0
  [935fb764] Xorg_libXcursor_jll v1.2.0+4
  [a3789734] Xorg_libXdmcp_jll v1.1.4+0
  [1082639a] Xorg_libXext_jll v1.3.4+4
  [d091e8ba] Xorg_libXfixes_jll v5.0.3+4
  [a51aa0fd] Xorg_libXi_jll v1.7.10+4
  [d1454406] Xorg_libXinerama_jll v1.1.4+4
  [ec84b674] Xorg_libXrandr_jll v1.5.2+4
  [ea2f1a96] Xorg_libXrender_jll v0.9.10+4
  [14d82f49] Xorg_libpthread_stubs_jll v0.1.1+0
  [c7cfdc94] Xorg_libxcb_jll v1.15.0+0
  [cc61e674] Xorg_libxkbfile_jll v1.1.2+0
  [e920d4aa] Xorg_xcb_util_cursor_jll v0.1.4+0
  [12413925] Xorg_xcb_util_image_jll v0.4.0+1
  [2def613f] Xorg_xcb_util_jll v0.4.0+1
  [975044d2] Xorg_xcb_util_keysyms_jll v0.4.0+1
  [0d47668e] Xorg_xcb_util_renderutil_jll v0.3.9+1
  [c22f9ab0] Xorg_xcb_util_wm_jll v0.4.1+1
  [35661453] Xorg_xkbcomp_jll v1.4.6+0
  [33bec58e] Xorg_xkeyboard_config_jll v2.39.0+0
  [c5fb5394] Xorg_xtrans_jll v1.5.0+0
  [3161d3a3] Zstd_jll v1.5.5+0
  [35ca27e7] eudev_jll v3.2.9+0
  [214eeab7] fzf_jll v0.43.0+0
  [1a1c6b14] gperf_jll v3.1.1+0
  [a4ae2306] libaom_jll v3.4.0+0
  [0ac62f75] libass_jll v0.15.1+0
  [2db6ffa8] libevdev_jll v1.11.0+0
  [f638f0a6] libfdk_aac_jll v2.0.2+0
  [36db933b] libinput_jll v1.18.0+0
  [b53b4c65] libpng_jll v1.6.40+0
  [f27f6e37] libvorbis_jll v1.3.7+1
  [009596ad] mtdev_jll v1.1.6+0
  [1270edf5] x264_jll v2021.5.5+0
  [dfaa095f] x265_jll v3.5.0+0
  [d8fb68d0] xkbcommon_jll v1.4.1+1
  [0dad84c5] ArgTools v1.1.1
  [56f22d72] Artifacts
  [2a0f44e3] Base64
  [ade2ca70] Dates
  [8ba89e20] Distributed
  [f43a241f] Downloads v1.6.0
  [7b1f6079] FileWatching
  [9fa8497b] Future
  [b77e0a4c] InteractiveUtils
  [4af54fe1] LazyArtifacts
  [b27032c2] LibCURL v0.6.4
  [76f85450] LibGit2
  [8f399da3] Libdl
  [37e2e46d] LinearAlgebra
  [56ddb016] Logging
  [d6f4376e] Markdown
  [a63ad114] Mmap
  [ca575930] NetworkOptions v1.2.0
  [44cfe95a] Pkg v1.10.0
  [de0858da] Printf
  [9abbd945] Profile
  [3fa0cd96] REPL
  [9a3f8284] Random
  [ea8e919c] SHA v0.7.0
  [9e88b42a] Serialization
  [1a1011a3] SharedArrays
  [6462fe0b] Sockets
  [2f01184e] SparseArrays v1.10.0
  [10745b16] Statistics v1.10.0
  [4607b0f0] SuiteSparse
  [fa267f1f] TOML v1.0.3
  [a4e569a6] Tar v1.10.0
  [8dfed614] Test
  [cf7118a7] UUIDs
  [4ec0a83e] Unicode
  [e66e0078] CompilerSupportLibraries_jll v1.0.5+1
  [781609d7] GMP_jll v6.2.1+6
  [deac9b47] LibCURL_jll v8.4.0+0
  [e37daf67] LibGit2_jll v1.6.4+0
  [29816b5a] LibSSH2_jll v1.11.0+1
  [3a97d323] MPFR_jll v4.2.0+1
  [c8ffd9c3] MbedTLS_jll v2.28.2+1
  [14a3606d] MozillaCACerts_jll v2023.1.10
  [4536629a] OpenBLAS_jll v0.3.23+2
  [05823500] OpenLibm_jll v0.8.1+2
  [efcefdf7] PCRE2_jll v10.42.0+1
  [bea87d4a] SuiteSparse_jll v7.2.1+1
  [83775a58] Zlib_jll v1.2.13+1
  [8e850b90] libblastrampoline_jll v5.8.0+1
  [8e850ede] nghttp2_jll v1.52.0+1
  [3f19e933] p7zip_jll v17.4.0+2
Julia Version 1.10.0
Commit 3120989f39b (2023-12-25 18:01 UTC)
Build Info:
  Official https://julialang.org/ release
Platform Info:
  OS: macOS (x86_64-apple-darwin22.4.0)
  CPU: 8 × Intel(R) Core(TM) i5-8279U CPU @ 2.40GHz
  WORD_SIZE: 64
  LIBM: libopenlibm
  LLVM: libLLVM-15.0.7 (ORCJIT, skylake)
  Threads: 11 on 8 virtual cores
Environment:
  JULIA_EDITOR = code

Additional context See https://discourse.julialang.org/t/why-does-my-cubature-of-odesolution-break-down-with-finite-limits/108736

Also, the problem in general seems to be non-existent when using a different cubature with using Cubature.

lxvm commented 9 months ago

@homocomputeris was this resolved on discourse?

homocomputeris commented 9 months ago

@homocomputeris was this resolved on discourse?

Chris's solution work for me but it doesn't fix the cubature itself. The fact that the cubature need some specific array type is still present.

Also

Open an issue on Integrals.jl so we can discuss this more.

lxvm commented 9 months ago

Looking into the MWE and discourse post, I don't think the issue is with Integrals.jl but with OrdinaryDiffEQ.jl. It is true that HCubature.jl tries to use StaticArrays.jl as the evaluation points, but the caller is responsible for making sure their integrand is compatible with that input. This won't be changed because it is currently the best way to get fast code for small integrands or systems of equations. As discussed on discourse, the patch to fix this is to simply make sure the ODE solver always sees a Vector, e.g. with prob = IntegralProblem((tail, p) -> jpdf(collect([x; tail]), p, t), ..., but the underlying problem is probably some mishandling of static array types in the function OrdinaryDiffEq.calc_W whose result disagrees with the pre-computed cache type used in OrdinaryDiffEq.update_W! that is causing the conversion error

ERROR: MethodError: Cannot `convert` an object of type LinearAlgebra.LU{Float64, Matrix{Float64}, Vector{Int64}} to an object of type StaticArrays.LU{LinearAlgebra.LowerTriangular{Float64, StaticArraysCore.SMatrix{3, 3, Float64, 9}}, LinearAlgebra.UpperTriangular{Float64, StaticArraysCore.SMatrix{3, 3, Float64, 9}}, StaticArraysCore.SVector{3, Int64}}

@ChrisRackauckas this seems like an OrdinaryDiffEq.jl bug, correct?

Also, I saw that Mwe.x1 is broken as well, and there is no difference between (in)finite limits.

homocomputeris commented 9 months ago

Also, I saw that Mwe.x1 is broken as well, and there is no difference between (in)finite limits.

With this code x1 works but x1_broken doesn't. It's probably due to rather stupid mistake AutoTsit5(Vern9()) (which was probably supposed to be AutoVern9(...) but the fact still holds: for some integration limits it works, for some it doesn't.

module Mwe

using DifferentialEquations
using Integrals
using Distributions
using LinearAlgebra
using ForwardDiff

# just to speed things up a bit
const TOL = 1e-1

# some parameters; exact values do not matter
tspan = (0, pi)
p = [1, 1]
t = 22 / 7
# limit the domain of interest
rectangle = (-5, 5)

# ODE somewhat similar to the harmonic oscillator
function f(du, u, p, t)
    du[1] = u[2]
    du[2] = -u[1]
    du[3] = 0.0
    return du # (or return nothing)
 end

# scalar function with vector argument
function g0(x, p, t)
    return pdf(truncated(Normal(1, 1), rectangle...), x[1]) *
           pdf(truncated(Normal(1, 1), rectangle...), x[2]) *
           pdf(truncated(Normal(1, 1), rectangle...), x[3])
end

function g1(x, p, t)
    # find the initial condition for a given point
    function tmp(y)
        prob = ODEProblem(f, y, reverse(tspan), p)
        sol = solve(prob, AutoTsit5(Vern9());
            abstol = TOL, reltol = TOL)
        return sol(tspan[end] - t)
    end
    # use the found IC in g0
    return g0(tmp(x), p, 0.0) * abs(det(ForwardDiff.jacobian(tmp, x)))
end

# Integrate everywhere
function x1(jpdf, x::Number, p, t)
    prob = IntegralProblem((tail, p) -> jpdf([x; tail], p, t),
        [-5, -Inf],
        [5, Inf], p)
    sol = solve(prob, HCubatureJL();
        reltol = TOL, abstol = TOL)
    return sol.u
end

# Truncate in a rectangle
function x1_broken(jpdf, x::Number, p, t)
    prob = IntegralProblem((tail, p) -> jpdf([x; tail], p, t),
        [-5, rectangle[1]],
        [5, rectangle[2]], p)
    sol = solve(prob, HCubatureJL();
        reltol = TOL, abstol = TOL)
    return sol.u
end

@show g0(rand(3), p, t)
@show g1(rand(3), p, t)

# this works
@show x1(g1, 3.0, p, t)

# this doesn't
@show x1_broken(g1, 3.0, p, t)

end
g0(rand(3), p, t) = 0.04281316658626982
g1(rand(3), p, t) = 0.0015088150936285213
x1(g1, 3.0, p, t) = 7.758772211499161e-5
ERROR: Initial condition incompatible with functional form.
Detected an in-place function with an initial condition of type Number or SArray.
This is incompatible because Numbers cannot be mutated, i.e.
`x = 2.0; y = 2.0; x .= y` will error.

If using a immutable initial condition type, please use the out-of-place form.
I.e. define the function `du=f(u,p,t)` instead of attempting to "mutate" the immutable `du`.

If your differential equation function was defined with multiple dispatches and one is
in-place, then the automatic detection will choose in-place. In this case, override the
choice in the problem constructor, i.e. `ODEProblem{false}(f,u0,tspan,p,kwargs...)`.

For a longer discussion on mutability vs immutability and in-place vs out-of-place, see:
https://diffeq.sciml.ai/stable/tutorials/faster_ode_example/#Example-Accelerating-a-Non-Stiff-Equation:-The-Lorenz-Equation

Some of the types have been truncated in the stacktrace for improved reading. To emit complete information
in the stack trace, evaluate `TruncatedStacktraces.VERBOSE[] = true` and re-run the code.

Stacktrace:
  [1] get_concrete_u0(prob::SciMLBase.ODEProblem{…}, isadapt::Bool, t0::Float64, kwargs::@Kwargs{…})
    @ DiffEqBase ~/.julia/packages/DiffEqBase/eLhx9/src/solve.jl:1292
  [2] get_concrete_problem(prob::SciMLBase.ODEProblem{…}, isadapt::Bool; kwargs::@Kwargs{…})
    @ DiffEqBase ~/.julia/packages/DiffEqBase/eLhx9/src/solve.jl:1148
  [3] solve_up(prob::SciMLBase.ODEProblem{…}, sensealg::Nothing, u0::StaticArraysCore.SVector{…}, p::Vector{…}, args::OrdinaryDiffEq.CompositeAlgorithm{…}; kwargs::@Kwargs{…})
    @ DiffEqBase ~/.julia/packages/DiffEqBase/eLhx9/src/solve.jl:1052
  [4] solve(prob::SciMLBase.ODEProblem{…}, args::OrdinaryDiffEq.CompositeAlgorithm{…}; sensealg::Nothing, u0::Nothing, p::Nothing, wrap::Val{…}, kwargs::@Kwargs{…})
    @ DiffEqBase ~/.julia/packages/DiffEqBase/eLhx9/src/solve.jl:981
  [5] (::Main.Mwe.var"#tmp#1"{Vector{Int64}, Float64})(y::StaticArraysCore.SVector{3, Float64})
    @ Main.Mwe ./Untitled-1:38
  [6] g1(x::StaticArraysCore.SVector{3, Float64}, p::Vector{Int64}, t::Float64)
    @ Main.Mwe ./Untitled-1:43
  [7] #4
    @ ./Untitled-1:58 [inlined]
  [8] IntegralFunction
    @ ~/.julia/packages/SciMLBase/aft1j/src/scimlfunctions.jl:2183 [inlined]
  [9] #46
    @ ~/.julia/packages/Integrals/d5Wr6/src/Integrals.jl:113 [inlined]
 [10] (::HCubature.GenzMalik{…})(f::Integrals.var"#46#48"{…}, a::StaticArraysCore.SVector{…}, b::StaticArraysCore.SVector{…}, norm::typeof(LinearAlgebra.norm))
    @ HCubature ~/.julia/packages/HCubature/QvyJW/src/genz-malik.jl:121
 [11] hcubature_(f::Integrals.var"#46#48"{…}, a::StaticArraysCore.SVector{…}, b::StaticArraysCore.SVector{…}, norm::typeof(LinearAlgebra.norm), rtol_::Float64, atol::Float64, maxevals::Int64, initdiv::Int64)
    @ HCubature ~/.julia/packages/HCubature/QvyJW/src/HCubature.jl:61
 [12] hcubature_(f::Integrals.var"#46#48"{…}, a::Vector{…}, b::Vector{…}, norm::Function, rtol::Float64, atol::Float64, maxevals::Int64, initdiv::Int64)
    @ HCubature ~/.julia/packages/HCubature/QvyJW/src/HCubature.jl:129
 [13] hcubature
    @ ~/.julia/packages/HCubature/QvyJW/src/HCubature.jl:178 [inlined]
 [14] __solvebp_call(prob::SciMLBase.IntegralProblem{…}, alg::Integrals.HCubatureJL{…}, sensealg::Integrals.ReCallVJP{…}, domain::Tuple{…}, p::Vector{…}; reltol::Float64, abstol::Float64, maxiters::Int64)
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/Integrals.jl:121
 [15] __solvebp_call
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/Integrals.jl:102 [inlined]
 [16] #__solvebp_call#4
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/common.jl:113 [inlined]
 [17] __solvebp_call
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/common.jl:112 [inlined]
 [18] __solvebp
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/Integrals.jl:65 [inlined]
 [19] solve!(cache::Integrals.IntegralCache{…})
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/common.jl:103
 [20] solve(prob::SciMLBase.IntegralProblem{…}, alg::Integrals.HCubatureJL{…}; kwargs::@Kwargs{…})
    @ Integrals ~/.julia/packages/Integrals/d5Wr6/src/common.jl:99
 [21] x1_broken(jpdf::typeof(Main.Mwe.g1), x::Float64, p::Vector{Int64}, t::Float64)
    @ Main.Mwe ./Untitled-1:61
 [22] macro expansion
    @ show.jl:1181 [inlined]
 [23] top-level scope
    @ Untitled-1:73
Some type information was truncated. Use `show(err)` to see complete types.
lxvm commented 9 months ago

With this code x1 works but x1_broken doesn't.

OK, it would probably be better to call this Mwe2, since there is a silent error, i.e. x1 should also break with HCubatureJL() for the same reason x1_broken does. It looks like the infinity transformation can change the type of the input point, which it shouldn't, so I will open a pr to correct that.

Now, your MWE makes it clear to me that you are expecting the type of the quadrature points to match the type of the limits of integration, however, you gave mutable limits of type Vector and instead are getting immutable points of type SVector. I agree it would be reasonable for Integrals.jl to enforce consistent types, so it would be good to add this to the list in #215.

ChrisRackauckas commented 9 months ago

@ChrisRackauckas this seems like an OrdinaryDiffEq.jl bug, correct?

No, the ODE solver is doing as expected. If u0 is a StaticArray, then your f should be returning StaticArrays. I think we could add an earlier error throw on this that's more explicit, but the issue is that u0 is a StaticArray but f is returning Vectors, which is not a behavior that is supported and something expected to error.

This originates though because HCubature is giving a user StaticArray values even though they never used static arrays, in which case they have to be very careful in their usage of this specific algorithm.

I agree it would be reasonable for Integrals.jl to enforce consistent types, so it would be good to add this to the list in https://github.com/SciML/Integrals.jl/issues/215.

Yes I think that's the key here. It's a bit odd for Integrals.jl to give the user StaticArray values without any warning on it. HCubature.jl does seem to document this (though I had no idea it did this before this issue was opened 😅), but Integrals.jl doesn't document this and it's the only solver that has this behavior. So it's a bit of an interface break and quite odd. In the wrapper we could create our own cache arrays, write into them, and send it to the user, and then have a keyword argument in the algorithm type to simply send the static array on to them, but I can see why it's confusing to a user to have this as the default behavior.