Merge pull request #106 from avik-pal/ap/nonvec

Towards GPU Compatible BVP Solvers
SciML · Sep 6, 2023 · a615b8a · a615b8a
2 parents 756e288 + b4ae596
commit a615b8a
Show file tree

Hide file tree

Showing 17 changed files with 528 additions and 297 deletions.
diff --git a/.JuliaFormatter.toml b/.JuliaFormatter.toml
@@ -1 +1,2 @@
-style = "sciml"
+style = "sciml"
+annotate_untyped_fields_with_any = false
diff --git a/Project.toml b/Project.toml
@@ -4,8 +4,9 @@ version = "4.1.0"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
+Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
-BandedMatrices = "aae01518-5342-5314-be14-df237901396f"
+ConcreteStructs = "2569d6c7-a4a2-43d3-a901-331e8e4be471"
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -21,8 +22,9 @@ UnPack = "3a884ed6-31ef-47d7-9d2a-63182c4928ed"
 
 [compat]
 ADTypes = "0.2"
+Adapt = "3"
 ArrayInterface = "7"
-BandedMatrices = "0.14, 0.15, 0.16, 0.17"
+ConcreteStructs = "0.2"
 DiffEqBase = "6.94.2"
 ForwardDiff = "0.10"
 NonlinearSolve = "1"

diff --git a/src/BoundaryValueDiffEq.jl b/src/BoundaryValueDiffEq.jl
@@ -1,18 +1,19 @@
 module BoundaryValueDiffEq
 
-using LinearAlgebra, PreallocationTools, Reexport, Setfield, SparseArrays
+using Adapt, LinearAlgebra, PreallocationTools, Reexport, Setfield, SparseArrays
 @reexport using ADTypes, DiffEqBase, NonlinearSolve, SparseDiffTools
 
 import ADTypes: AbstractADType
-import ArrayInterface: matrix_colors
-import BandedMatrices: BandedMatrix
+import ArrayInterface: matrix_colors, parameterless_type
+import ConcreteStructs: @concrete
 import DiffEqBase: solve
 import ForwardDiff: pickchunksize
 import SparseDiffTools: AbstractSparseADType
 import TruncatedStacktraces: @truncate_stacktrace
 import UnPack: @unpack
 
 include("types.jl")
+include("utils.jl")
 include("algorithms.jl")
 include("alg_utils.jl")
 include("mirk_tableaus.jl")