We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b34fd8f commit e812893Copy full SHA for e812893
test/gpu.jl
@@ -1,4 +1,15 @@
1
-using RecursiveArrayTools, CuArrays
+using RecursiveArrayTools, Adapt, CuArrays, Test
2
a = ArrayPartition(([1.0f0] |> cu, [2.0f0] |> cu, [3.0f0] |> cu))
3
b = ArrayPartition(([0.0f0] |> cu, [0.0f0] |> cu, [0.0f0] |> cu))
4
@. a + b
5
+
6
+a = VectorOfArray([ones(2) for i in 1:3])
7
+_a = Adapt.adapt(CuArray,a)
8
+@test _a isa VectorOfArray
9
+@test _a.u isa Vector{<:CuArray}
10
11
+b = DiffEqArray([ones(2) for i in 1:3],ones(2))
12
+_b = Adapt.adapt(CuArray,b)
13
+@test _b isa DiffEqArray
14
+@test _b.u isa Vector{<:CuArray}
15
+@test _b.t isa CuArray
0 commit comments