diff --git a/julia/tests/DistributedTest.jl b/julia/tests/DistributedTest.jl index a9234d2..b10daf9 100644 --- a/julia/tests/DistributedTest.jl +++ b/julia/tests/DistributedTest.jl @@ -1,4 +1,5 @@ using BenchmarkTools +using ClusterManagers using CSV using DataFrames using Distributed @@ -10,7 +11,13 @@ using .TUG # 1. Environment Setup function setup_environment(num_procs::Int) if num_procs > 0 - added_procs = addprocs(num_procs) + if haskey(ENV, "SLURM_JOB_ID") + # Add SLURM processes + added_procs = addprocs(SlurmManager(num_procs), exclusive="") + else + # Add local processes + added_procs = addprocs(num_procs) + end # Use remotecall to include TUG on each new worker for proc in added_procs