forked from rarias/bscpkgs
Add OSU test benchmarks
This commit is contained in:
@@ -18,13 +18,17 @@ let
|
||||
config = {
|
||||
cc = [ bsc.icc ];
|
||||
blocksize = [ 2048 ];
|
||||
mpi = [ bsc.impi bsc.openmpi bsc.mpich ];
|
||||
};
|
||||
|
||||
extraConfig = {
|
||||
particles = 16384;
|
||||
particles = 32*1024;
|
||||
timesteps = 10;
|
||||
ntasks = 2;
|
||||
mpi = bsc.impi;
|
||||
ntasksPerNode = 2;
|
||||
nodes = 1;
|
||||
time = "00:10:00";
|
||||
qos = "debug";
|
||||
#mpi = bsc.impi;
|
||||
#mpi = bsc.openmpi;
|
||||
gitBranch = "garlic/mpi+send";
|
||||
gitURL = "ssh://git@bscpm02.bsc.es/garlic/apps/nbody.git";
|
||||
@@ -37,7 +41,10 @@ let
|
||||
app = app;
|
||||
nixPrefix = "/gpfs/projects/bsc15/nix";
|
||||
exclusive = false;
|
||||
ntasks = "${toString conf.ntasks}";
|
||||
ntasksPerNode = "${toString conf.ntasksPerNode}";
|
||||
nodes = "${toString conf.nodes}";
|
||||
time = conf.time;
|
||||
qos = conf.qos;
|
||||
chdirPrefix = "/home/bsc15/bsc15557/bsc-nixpkgs/out";
|
||||
};
|
||||
|
||||
@@ -61,7 +68,7 @@ let
|
||||
nbody.override { inherit cc mpi blocksize gitBranch gitURL; };
|
||||
|
||||
pipeline = conf:
|
||||
# sbatch conf (
|
||||
sbatch conf (
|
||||
srun (
|
||||
nixsetupWrapper (
|
||||
argv conf (
|
||||
@@ -69,7 +76,7 @@ let
|
||||
)
|
||||
)
|
||||
)
|
||||
# )
|
||||
)
|
||||
;
|
||||
|
||||
# Ideally it should look like this:
|
||||
|
||||
70
bsc/garlic/exp/osu/latency.nix
Normal file
70
bsc/garlic/exp/osu/latency.nix
Normal file
@@ -0,0 +1,70 @@
|
||||
{
|
||||
bsc
|
||||
, genApp
|
||||
, genConfigs
|
||||
|
||||
# Wrappers
|
||||
, launchWrapper
|
||||
, sbatchWrapper
|
||||
, srunWrapper
|
||||
, argvWrapper
|
||||
, controlWrapper
|
||||
, nixsetupWrapper
|
||||
}:
|
||||
|
||||
let
|
||||
# Set the configuration for the experiment
|
||||
config = {
|
||||
mpi = [ bsc.impi bsc.openmpi bsc.mpich ];
|
||||
};
|
||||
|
||||
extraConfig = {
|
||||
ntasksPerNode = 1;
|
||||
nodes = 2;
|
||||
time = "00:10:00";
|
||||
qos = "debug";
|
||||
};
|
||||
|
||||
# Compute the cartesian product of all configurations
|
||||
configs = map (conf: conf // extraConfig) (genConfigs config);
|
||||
|
||||
sbatch = conf: app: sbatchWrapper {
|
||||
app = app;
|
||||
nixPrefix = "/gpfs/projects/bsc15/nix";
|
||||
exclusive = false;
|
||||
ntasksPerNode = "${toString conf.ntasksPerNode}";
|
||||
nodes = "${toString conf.nodes}";
|
||||
time = conf.time;
|
||||
qos = conf.qos;
|
||||
chdirPrefix = "/home/bsc15/bsc15557/bsc-nixpkgs/out";
|
||||
};
|
||||
|
||||
srun = app: srunWrapper {
|
||||
app = app;
|
||||
nixPrefix = "/gpfs/projects/bsc15/nix";
|
||||
};
|
||||
|
||||
argv = app:
|
||||
argvWrapper {
|
||||
app = app;
|
||||
program = "bin/osu_latency";
|
||||
argv = "()";
|
||||
env = ''
|
||||
export I_MPI_THREAD_SPLIT=1
|
||||
'';
|
||||
};
|
||||
|
||||
osumbFn = conf:
|
||||
with conf;
|
||||
bsc.osumb.override { inherit mpi; };
|
||||
|
||||
pipeline = conf: srun (nixsetupWrapper (argv (osumbFn conf)));
|
||||
#pipeline = conf: sbatch conf (srun (nixsetupWrapper (argv bsc.osumb)));
|
||||
|
||||
# Ideally it should look like this:
|
||||
#pipeline = sbatch nixsetup control argv nbodyFn;
|
||||
|
||||
jobs = map pipeline configs;
|
||||
|
||||
in
|
||||
launchWrapper jobs
|
||||
Reference in New Issue
Block a user