Skip to content

Commit 183413b

Browse files
committed
tests: Support other GPU vendors
1 parent be1bfb4 commit 183413b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

test/runtests.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,16 @@
11
using PSOGPU
22
using Test, StaticArrays, LinearAlgebra, Random
33

4-
const GROUP = get(ENV, "GROUP", "All")
4+
const GROUP = get(ENV, "GROUP", "CPU")
55

66
@testset "Rosenbrock test dimension = $(n)" for n in 2:4
77
global N = n
88
include("./regression.jl")
99
end
1010

11-
if GROUP == "CUDA"
11+
if GROUP != "CPU"
12+
@eval using $(Symbol(GROUP))
13+
1214
@testset "Rosenbrock on gpu" begin
1315
include("./gpu.jl")
1416
end

0 commit comments

Comments
 (0)