format

Author: utkarsh530

benchmarks/CPU_vs_GPU/benchmark.jl

@@ -205,8 +205,7 @@ plt = plot(Ns,
     xticks = xticks,
     yticks = yticks,
     marker = :circle,
-    dpi = 600
-    # color = :Green
+    dpi = 600    # color = :Green
 )
 
 plt = plot!(Ns,
@@ -216,8 +215,7 @@ plt = plot!(Ns,
     linewidth = 2,
     label = "ParallelSyncPSOKernel: CPU",
     marker = :circle,
-    ls = :dash
-    # color = :Orange
+    ls = :dash    # color = :Orange
 )
 
 plt = plot!(Ns,
@@ -226,8 +224,7 @@ plt = plot!(Ns,
     yaxis = :log,
     linewidth = 2,
     label = "ParallelPSOKernel (Async): GPU",
-    marker = :circle
-    # color = :Green
+    marker = :circle    # color = :Green
 )
 
 plt = plot!(Ns,
@@ -236,8 +233,7 @@ plt = plot!(Ns,
     yaxis = :log,
     linewidth = 2,
     label = "ParallelPSOKernel (Queue-lock): GPU",
-    marker = :circle
-    # color = :Green
+    marker = :circle    # color = :Green
 )
 
 plt = plot!(Ns,
@@ -246,8 +242,7 @@ plt = plot!(Ns,
     yaxis = :log,
     linewidth = 2,
     label = "HybridPSO-LBFGS: GPU",
-    marker = :circle
-    # color = :Green
+    marker = :circle    # color = :Green
 )
 
 savefig("benchmark_hybrid.svg")

benchmarks/CPU_vs_GPU/wp_algs.jl

@@ -167,8 +167,7 @@ plt = plot(gpu_sync_times,
     xticks = xticks,
     yticks = yticks,
     marker = :circle,
-    dpi = 600
-    # color = :Green
+    dpi = 600    # color = :Green
 )
 
 plt = plot!(cpu_times,
@@ -187,8 +186,7 @@ plt = plot!(gpu_async_times,
     yaxis = :log,
     linewidth = 2,
     label = "ParallelPSOKernel (Async): GPU",
-    marker = :circle
-    # color = :Green
+    marker = :circle    # color = :Green
 )
 
 plt = plot!(gpu_queue_lock_times,
@@ -197,8 +195,7 @@ plt = plot!(gpu_queue_lock_times,
     yaxis = :log,
     linewidth = 2,
     label = "ParallelPSOKernel (Queue-lock): GPU",
-    marker = :circle
-    # color = :Green
+    marker = :circle    # color = :Green
 )
 
 plt = plot!(gpu_hybrid_times,
@@ -207,8 +204,7 @@ plt = plot!(gpu_hybrid_times,
     yaxis = :log,
     linewidth = 2,
     label = "HybridPSO-LBFGS: GPU",
-    marker = :circle
-    # color = :Green
+    marker = :circle    # color = :Green
 )
 
 savefig("wp_pso.svg")
@@ -434,8 +430,7 @@ plt = plot(lbfgs_time,
     xticks = xticks,
     yticks = yticks,
     marker = :circle,
-    dpi = 600
-    # color = :Green
+    dpi = 600    # color = :Green
 )
 
 plt = plot!(adam_time,
@@ -444,8 +439,7 @@ plt = plot!(adam_time,
     yaxis = :log,
     linewidth = 2,
     label = "ADAM",
-    marker = :circle
-    # color = :Orange
+    marker = :circle    # color = :Orange
 )
 
 plt = plot!(pso_cpu_time,
@@ -454,8 +448,7 @@ plt = plot!(pso_cpu_time,
     # yaxis = :log,
     linewidth = 2,
     label = "ParallelSyncPSOKernel: CPU",
-    marker = :circle
-    # color = :Green
+    marker = :circle    # color = :Green
 )
 
 plt = plot!(queue_lock_time,
@@ -464,8 +457,7 @@ plt = plot!(queue_lock_time,
     yaxis = :log,
     linewidth = 2,
     label = "ParallelPSOKernel (queue-lock): GPU",
-    marker = :circle
-    # color = :Orange
+    marker = :circle    # color = :Orange
 )
 
 plt = plot!(hybrid_time,
@@ -474,8 +466,7 @@ plt = plot!(hybrid_time,
     yaxis = :log,
     linewidth = 2,
     label = "HybridPSO-LBFGS: GPU",
-    marker = :circle
-    # color = :Orange
+    marker = :circle    # color = :Orange
 )
 
 plt = plot!(bbo_time,
@@ -484,8 +475,7 @@ plt = plot!(bbo_time,
     # yaxis = :log,
     linewidth = 2,
     label = "BlackboxOptim",
-    marker = :circle
-    # color = :Green
+    marker = :circle    # color = :Green
 )
 
 savefig("wp_algs.svg")

benchmarks/Fitzhugh_Nagumo/ode_estimation.jl

@@ -61,7 +61,8 @@ ub = @SArray fill(5.0f0, 4)
 
 u_guess = @MArray zeros(Float32, 4)
 
-optprob = OptimizationProblem(Optimization.OptimizationFunction(loss,
+optprob = OptimizationProblem(
+    Optimization.OptimizationFunction(loss,
         Optimization.AutoForwardDiff()),
     u_guess,
     (prob_short, t_short))

benchmarks/Lotka_Volterra/lotka_volterra.jl

@@ -53,7 +53,8 @@ optprob = OptimizationProblem(loss, opt_u0, (prob_short, t_short); lb = lb, ub =
 
 u_guess = @MArray zeros(Float64, 4)
 
-optprob = OptimizationProblem(Optimization.OptimizationFunction(loss,
+optprob = OptimizationProblem(
+    Optimization.OptimizationFunction(loss,
         Optimization.AutoForwardDiff()),
     u_guess,
     (prob_short, t_short))

benchmarks/NeuralODE/adam_opt.jl

@@ -3,7 +3,7 @@ using Pkg
 Pkg.activate(@__DIR__)
 
 using SimpleChains,
-    StaticArrays, OrdinaryDiffEq, SciMLSensitivity, Optimization, OptimizationFlux, Plots
+      StaticArrays, OrdinaryDiffEq, SciMLSensitivity, Optimization, OptimizationFlux, Plots
 
 using CUDA
 device!(2)

benchmarks/adam_opt.jl

@@ -1,5 +1,5 @@
 using SimpleChains,
-    StaticArrays, OrdinaryDiffEq, SciMLSensitivity, Optimization, OptimizationFlux, Plots
+      StaticArrays, OrdinaryDiffEq, SciMLSensitivity, Optimization, OptimizationFlux, Plots
 
 #device!(2)
 # Get Tesla V100S

src/PSOGPU.jl

@@ -68,5 +68,5 @@ include("./bfgs.jl")
 include("./hybrid.jl")
 
 export ParallelPSOKernel,
-    ParallelSyncPSOKernel, ParallelPSOArray, SerialPSO, OptimizationProblem, solve
+       ParallelSyncPSOKernel, ParallelPSOArray, SerialPSO, OptimizationProblem, solve
 end