Skip to content

Commit ebf1279

Browse files
cuda backend
Signed-off-by: AdityaPandeyCN <adityapand3y666@gmail.com>
1 parent 98f1fd7 commit ebf1279

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

benchmarks/PSOGlobalOptimization/pso_global_optimizers.jmd

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,15 @@ using OptimizationMetaheuristics, OptimizationSciPy
2727
using ParallelParticleSwarms
2828
using ForwardDiff
2929
using KernelAbstractions
30+
using CUDA
3031
using StaticArrays, LinearAlgebra
3132

3233
const PSOKernel = ParallelParticleSwarms.ParallelPSOKernel
3334
const SyncPSOKernel = ParallelParticleSwarms.ParallelSyncPSOKernel
3435
const SerPSO = ParallelParticleSwarms.SerialPSO
3536
const HPso = ParallelParticleSwarms.HybridPSO
3637

37-
const BACKEND = CPU() # use CUDABackend() from CUDA.jl on GPU CI
38+
const BACKEND = CUDABackend()
3839
```
3940

4041
```julia

0 commit comments

Comments
 (0)