Skip to content

Commit d1509e6

Browse files
committed
Allow scalar for comparison
1 parent 2afc31d commit d1509e6

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

test/gpu/cuda.jl

+6-4
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
using LinearSolve, CUDA, LinearAlgebra, SparseArrays
22
using Test
33

4+
CUDA.allowscalar(false)
5+
46
n = 8
57
A = Matrix(I, n, n)
68
b = ones(n)
@@ -25,19 +27,19 @@ function test_interface(alg, prob1, prob2)
2527
x2 = prob2.u0
2628

2729
y = solve(prob1, alg; cache_kwargs...)
28-
@test Array(A1 * y) Array(b1)
30+
@test CUDA.@allowscalar(Array(A1 * y) Array(b1))
2931

3032
cache = SciMLBase.init(prob1, alg; cache_kwargs...) # initialize cache
3133
solve!(cache)
32-
@test Array(A1 * cache.u) Array(b1)
34+
@test CUDA.@allowscalar(Array(A1 * cache.u) Array(b1))
3335

3436
cache.A = copy(A2)
3537
solve!(cache)
36-
@test Array(A2 * cache.u) Array(b1)
38+
@test CUDA.@allowscalar(Array(A2 * cache.u) Array(b1))
3739

3840
cache.b = copy(b2)
3941
solve!(cache)
40-
@test Array(A2 * cache.u) Array(b2)
42+
@test CUDA.@allowscalar(Array(A2 * cache.u) Array(b2))
4143

4244
return
4345
end

0 commit comments

Comments
 (0)