Add OSU test benchmarks

This commit is contained in:
2020-08-18 18:28:30 +02:00
parent ecc01e4314
commit 1e07be863a
7 changed files with 134 additions and 10 deletions

View File

@@ -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:

View 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