82 lines
1.5 KiB
Nix
82 lines
1.5 KiB
Nix
{
|
|
bsc
|
|
, nbody
|
|
, genApp
|
|
, genConfigs
|
|
|
|
# Wrappers
|
|
, launchWrapper
|
|
, sbatchWrapper
|
|
, srunWrapper
|
|
, argvWrapper
|
|
, controlWrapper
|
|
, nixsetupWrapper
|
|
}:
|
|
|
|
let
|
|
# Set the configuration for the experiment
|
|
config = {
|
|
cc = [ bsc.icc ];
|
|
blocksize = [ 2048 ];
|
|
};
|
|
|
|
extraConfig = {
|
|
particles = 16384;
|
|
timesteps = 10;
|
|
ntasks = 2;
|
|
mpi = bsc.impi;
|
|
#mpi = bsc.openmpi;
|
|
gitBranch = "garlic/mpi+send";
|
|
gitURL = "ssh://git@bscpm02.bsc.es/garlic/apps/nbody.git";
|
|
};
|
|
|
|
# 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;
|
|
ntasks = "${toString conf.ntasks}";
|
|
chdirPrefix = "/home/bsc15/bsc15557/bsc-nixpkgs/out";
|
|
};
|
|
|
|
srun = app: srunWrapper {
|
|
app = app;
|
|
nixPrefix = "/gpfs/projects/bsc15/nix";
|
|
};
|
|
|
|
argv = conf: app:
|
|
with conf;
|
|
argvWrapper {
|
|
app = app;
|
|
argv = ''(-t ${toString timesteps} -p ${toString particles})'';
|
|
env = ''
|
|
export I_MPI_THREAD_SPLIT=1
|
|
'';
|
|
};
|
|
|
|
nbodyFn = conf:
|
|
with conf;
|
|
nbody.override { inherit cc mpi blocksize gitBranch gitURL; };
|
|
|
|
pipeline = conf:
|
|
# sbatch conf (
|
|
srun (
|
|
nixsetupWrapper (
|
|
argv conf (
|
|
nbodyFn conf
|
|
)
|
|
)
|
|
)
|
|
# )
|
|
;
|
|
|
|
# Ideally it should look like this:
|
|
#pipeline = sbatch nixsetup control argv nbodyFn;
|
|
|
|
jobs = map pipeline configs;
|
|
|
|
in
|
|
launchWrapper jobs
|