Use JLArrays for gpu tests

JuliaGaussianProcesses · Feb 21, 2024 · 4468cd9 · 4468cd9
1 parent 4caf2d9
commit 4468cd9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/test/Project.toml b/test/Project.toml
@@ -1,2 +1,3 @@
 [deps]
+JLArrays = "27aeb0d3-9eb9-45fb-866b-73c2ecf80fcb"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,4 +1,5 @@
 using ScalarKernelFunctions
+using JLArrays
 using Test
 
 @testset "ScalarKernelFunctions" begin
@@ -52,9 +53,9 @@ using Test
             end
         end
     end
-    @testset "GPU element type and type stability" begin
-        x0 = rand(Float32, 10)
-        x1 = rand(Float32, 10)
+    @testset "GPU compatibility" begin
+        x0 = rand(Float32, 10) |> jl
+        x1 = rand(Float32, 10) |> jl
         @testset "$k" for k in (
             ScalarSEKernel(), ScalarLinearKernel(), ScalarPeriodicKernel()
         )