Skip to content

Commit 7446973

Browse files
authored
additional gpu support tests (#256)
1 parent 41ca636 commit 7446973

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

test/runtests.jl

+6-3
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ using TypedTables: Table
77
using DataAPI: refarray, refvalue
88
using Adapt: adapt, Adapt
99
using JLArrays
10+
using GPUArraysCore: backend
1011
using Test
1112
using SparseArrays
1213

@@ -1305,11 +1306,13 @@ Base.BroadcastStyle(::Broadcast.ArrayStyle{MyArray2}, S::Broadcast.DefaultArrayS
13051306
bcmul2(a) = 2 .* a
13061307
a = StructArray(randn(ComplexF32, 10, 10))
13071308
sa = jl(a)
1308-
backend = StructArrays.GPUArraysCore.backend
1309-
@test @inferred(backend(sa)) === backend(sa.re) === backend(sa.im)
1309+
@test sa isa StructArray
1310+
@test @inferred(backend(sa)) === backend(sa.re) === backend(sa.im) === backend(jl(a.re))
13101311
@test collect(@inferred(bcabs(sa))) == bcabs(a)
1312+
@test backend(bcabs(sa)) === backend(sa)
13111313
@test @inferred(bcmul2(sa)) isa StructArray
1312-
@test (sa .+= 1) isa StructArray
1314+
@test backend(bcmul2(sa)) === backend(sa)
1315+
@test (sa .+= 1) === sa
13131316
end
13141317
end
13151318

0 commit comments

Comments
 (0)