fixup! add backend variable

Author: utkarsh530

test/Project.toml

@@ -1,4 +1,5 @@
 [deps]
+CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"

test/gpu.jl

@@ -1,3 +1,5 @@
+include("./utils.jl")
+
 Random.seed!(1234)
 
 ## Solving the rosenbrock problem
@@ -16,12 +18,12 @@ prob = OptimizationProblem(rosenbrock, x0, p; lb = lb, ub = ub)
 
 n_particles = 1000
 
-sol = solve(prob, ParallelPSOKernel(n_particles; backend = CUDABackend()), maxiters = 500)
+sol = solve(prob, ParallelPSOKernel(n_particles; backend), maxiters = 500)
 
 @test sol.objective < 1e-4
 
 sol = solve(prob,
-    ParallelSyncPSO(n_particles, CUDABackend()),
+    ParallelSyncPSO(n_particles, backend),
     maxiters = 500)
 
 @test sol.objective < 1e-4