Skip to content

Commit e7bb17d

Browse files
committed
Wireup for MPI not yet working
1 parent c5c5e10 commit e7bb17d

File tree

2 files changed

+12
-11
lines changed

2 files changed

+12
-11
lines changed

benchmarks/Manifest.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# This file is machine-generated - editing it directly is not advised
22

3-
julia_version = "1.7.0-beta3.0"
3+
julia_version = "1.7.0-rc1"
44
manifest_format = "2.0"
55

66
[[deps.ArgTools]]

benchmarks/run.jl

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -19,15 +19,16 @@ function __set_io_path(spec::FluxRM.JobSpec.Jobspec, iotype, name, path)
1919
io["path"] = path
2020
end
2121

22-
function juliaspec(args, dir; num_nodes=1, num_tasks_per_node=6, cores_per_task=6)
22+
function juliaspec(args, dir; num_nodes=1, num_tasks_per_node=8, cores_per_task=1)
2323
num_tasks = num_nodes*num_tasks_per_node
2424
cmd = `$(Base.julia_cmd()) $(args)`
2525
jobspec = FluxRM.JobSpec.from_command(cmd; num_nodes, num_tasks, cores_per_task)
2626
system = jobspec.attributes.system
2727
system.cwd = dir
2828
system.environment = Dict(
29-
"JULIA_PROJECT" => dir,
30-
"OPENBLAS_NUM_THREADS" => "8" # HyperThreads
29+
"JULIA_PROJECT" => dir,
30+
"JULIA_NUM_THREADS" => cores_per_task,
31+
"JULIA_EXCLUSIVE" => 1
3132
)
3233
__set_io_path(jobspec, "output", "stderr", "flux-{{id}}.err")
3334
__set_io_path(jobspec, "output", "stdout", "flux-{{id}}.out")
@@ -50,16 +51,16 @@ function nodes()
5051
end
5152

5253
const N = nodes()
54+
const workdir = realpath(joinpath(@__DIR__, "..", "examples"))
5355

54-
let flux = Flux()
56+
@info "Launching Jobs in " workdir
5557

58+
let flux = Flux()
5659
for i in 0:floor(Int,log2(N))
5760
n = 2^i
58-
for psize in (20,)
59-
jobspec = juliaspec(`-L setup.jl experiment.jl $psize`, realpath("experiment"), num_nodes=n)
60-
sub = FluxRM.submit(flux, jobspec)
61-
job = FluxRM.Job(sub)
62-
@info "Launched" jobid = FluxRM.encode(job) n psize
63-
end
61+
jobspec = juliaspec(`benchmark.jl -s 45`, workdir, num_nodes=n)
62+
sub = FluxRM.submit(flux, jobspec)
63+
job = FluxRM.Job(sub)
64+
@info "Launched" jobid = FluxRM.encode(job) n
6465
end
6566
end

0 commit comments

Comments
 (0)