Skip to content

Commit 72fbbf4

Browse files
Update arraypartition_gpu.jl
1 parent 2419648 commit 72fbbf4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

test/gpu/arraypartition_gpu.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,7 @@ RecursiveArrayTools.recursivefill!(pA, true)
1717
# Test that regular filling is done using GPU kernels and not scalar indexing
1818
fill!(pA, false)
1919
@test all(pA .== false)
20+
21+
a = ArrayPartition(([1.0f0] |> cu, [2.0f0] |> cu, [3.0f0] |> cu))
22+
b = ArrayPartition(([0.0f0] |> cu, [0.0f0] |> cu, [0.0f0] |> cu))
23+
@. a + b

0 commit comments

Comments
 (0)