test

Author: xtalax

test/gpu/Project.toml

@@ -1,6 +1,11 @@
 [deps]
+Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
+FastBroadcast = "7034ab61-46d4-4ed7-9d0f-46aef9175898"
+FillArrays = "1a297f60-69ca-5386-bcde-b61e274b549b"
 OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
+RecursiveArrayTools = "731186ca-8d62-57ce-b412-fbd966d074cd"
 
 [compat]
 CUDA = "5"

test/gpu/hermite_test.jl

@@ -0,0 +1,15 @@
+using ComponentArrays, CUDA, Adapt, RecursiveArrayTools, FastBroadcast, FillArrays, OrdinaryDiffEq, Test
+
+a = ComponentArray((a=rand(Float32, 5,5), b=rand(Float32, 5, 5)))
+a = adapt(CuArray, a)
+pa = ArrayPartition(a)
+pb = deepcopy(pa)
+pc = deepcopy(pa)
+pd = deepcopy(pa)
+pe = deepcopy(pa)
+k = [pd, pe]
+t = FillArrays.Trues(length(pa))
+
+OrdinaryDiffEq.hermite_interpolant!(pa, 0.1, 0.2, pb, pc, k, nothing, Val{0}, t) # if this doesnt error we're good
+
+@test pa.a != pb.a

test/runtests.jl

@@ -192,5 +192,6 @@ end
         @time @safetestset "Autoswitch GPU" include("gpu/autoswitch.jl")
         @time @safetestset "Linear LSRK GPU" include("gpu/linear_lsrk.jl")
         @time @safetestset "Reaction-Diffusion Stiff Solver GPU" include("gpu/reaction_diffusion_stiff.jl")
+        @time @safetestset "Scalar indexing bug bypass" include("gpu/hermite_test.jl")
     end
 end # @time