Merge pull request #293 from JuliaDynamics/hw/fixgpu

fix gpu compat

Author: hexaeder

Project.toml

@@ -80,7 +80,7 @@ RecursiveArrayTools = "3.27.0"
 RuntimeGeneratedFunctions = "0.5.15"
 SciMLBase = "2"
 SparseArrays = "1"
-SparseConnectivityTracer = "0.6"
+SparseConnectivityTracer = "0.6, 1"
 Static = "1.1.1"
 StaticArrays = "1.9.4"
 SteadyStateDiffEq = "2.2.0"

docs/Project.toml

@@ -41,6 +41,7 @@ NetworkDynamicsInspector = {path = "../NetworkDynamicsInspector"}
 [compat]
 Bonito = "≥0.0.1"
 CairoMakie = "0.15"
+Chairmarks = "1.3.1"
 DataFrames = "≥0.0.1"
 DataInterpolations = "7, 8"
 DiffEqCallbacks = "4.2.2"
@@ -65,6 +66,7 @@ Printf = "≥0.0.1"
 Random = "≥0.0.1"
 SciMLBase = "≥0.0.1"
 SimpleWeightedGraphs = "1.4.0"
+SparseConnectivityTracer = "≥0.0.1"
 StableRNGs = "1.0.2"
 StaticArrays = "≥0.0.1"
 StyledStrings = "≥0.0.1"

ext/NetworkDynamicsCUDAExt.jl

@@ -39,7 +39,7 @@ function Adapt.adapt_structure(to, n::Network)
     extmap = adapt(to, n.extmap)
 
     Network{exT,gT,typeof(layer),typeof(vb),typeof(mm),eltype(caches),typeof(gbp),typeof(extmap)}(
-        vb, layer, n.im, caches, mm, gbp, extmap)
+        vb, layer, n.im, caches, mm, gbp, extmap, getfield(n, :jac_prototype))
 end
 
 Adapt.@adapt_structure NetworkLayer

test/Project.toml

@@ -69,6 +69,7 @@ DiffEqCallbacks = "4.1.0"
 DifferentiationInterface = "0.6, 0.7"
 DifferentiationInterfaceTest = "0.10"
 DynamicQuantities = "1.4.0"
+Enzyme = "0.13"
 ExplicitImports = "1.10.1"
 FiniteDiff = "2.26.2"
 ForwardDiff = "≥0.0.1"
@@ -79,6 +80,7 @@ KernelAbstractions = "≥0.0.1"
 LinearAlgebra = "≥0.0.1"
 Logging = "1"
 ModelingToolkit = "≥0.0.1"
+Mooncake = "0.4"
 NetworkDynamics = "≥0.0.1"
 OrderedCollections = "1.7.0"
 OrdinaryDiffEqNonlinearSolve = "1.3.0"
@@ -89,6 +91,7 @@ Pkg = "1"
 Plots = "1.40.9"
 Printf = "≥0.0.1"
 Random = "≥0.0.1"
+ReverseDiff = "1.16"
 SafeTestsets = "0.1.0"
 SciMLBase = "≥0.0.1"
 SimpleWeightedGraphs = "1.4.0"