From d1509e6586eb6ff0ca0f63def5b959b4a0ea60e6 Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@mit.edu>
Date: Sun, 5 Nov 2023 16:34:08 -0500
Subject: [PATCH] Allow scalar for comparison

---
 test/gpu/cuda.jl | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/test/gpu/cuda.jl b/test/gpu/cuda.jl
index 3d7070b5d..d0aaf579f 100644
--- a/test/gpu/cuda.jl
+++ b/test/gpu/cuda.jl
@@ -1,6 +1,8 @@
 using LinearSolve, CUDA, LinearAlgebra, SparseArrays
 using Test
 
+CUDA.allowscalar(false)
+
 n = 8
 A = Matrix(I, n, n)
 b = ones(n)
@@ -25,19 +27,19 @@ function test_interface(alg, prob1, prob2)
     x2 = prob2.u0
 
     y = solve(prob1, alg; cache_kwargs...)
-    @test Array(A1 * y) ≈ Array(b1)
+    @test CUDA.@allowscalar(Array(A1 * y) ≈ Array(b1))
 
     cache = SciMLBase.init(prob1, alg; cache_kwargs...) # initialize cache
     solve!(cache)
-    @test Array(A1 * cache.u) ≈ Array(b1)
+    @test CUDA.@allowscalar(Array(A1 * cache.u) ≈ Array(b1))
 
     cache.A = copy(A2)
     solve!(cache)
-    @test Array(A2 * cache.u) ≈ Array(b1)
+    @test CUDA.@allowscalar(Array(A2 * cache.u) ≈ Array(b1))
 
     cache.b = copy(b2)
     solve!(cache)
-    @test Array(A2 * cache.u) ≈ Array(b2)
+    @test CUDA.@allowscalar(Array(A2 * cache.u) ≈ Array(b2))
 
     return
 end