Add exp.nbody.tampi variants

This commit is contained in:
Rodrigo Arias 2020-10-21 18:43:40 +02:00
parent 7852d86a3f
commit 06c29b573f
2 changed files with 8 additions and 8 deletions

View File

@ -8,6 +8,7 @@
# Leave the first CPU per socket unused? # Leave the first CPU per socket unused?
, freeCpu ? false , freeCpu ? false
, particles ? 1024 * 32
}: }:
with stdenv.lib; with stdenv.lib;
@ -25,7 +26,7 @@ let
genConf = with bsc; c: targetMachine.config // rec { genConf = with bsc; c: targetMachine.config // rec {
inherit (machineConfig) hw; inherit (machineConfig) hw;
# nbody options # nbody options
particles = 1024 * 32; inherit particles;
timesteps = 10; timesteps = 10;
inherit (c) nblocks; inherit (c) nblocks;
totalTasks = ntasksPerNode * nodes; totalTasks = ntasksPerNode * nodes;

View File

@ -282,13 +282,12 @@ let
nbody = rec { nbody = rec {
test = callPackage ./garlic/exp/nbody/test.nix { }; test = callPackage ./garlic/exp/nbody/test.nix { };
tampi = callPackage ./garlic/exp/nbody/tampi.nix { }; tampi = callPackage ./garlic/exp/nbody/tampi.nix { };
baseline = tampi;
freeCpu = callPackage ./garlic/exp/nbody/tampi.nix { # Experiment variants
freeCpu = true; medium = tampi.override { particles = 64 * 1024; };
}; baseline = medium;
jemalloc = callPackage ./garlic/exp/nbody/tampi.nix { freeCpu = baseline.override { freeCpu = true; };
enableJemalloc = true; jemalloc = baseline.override { enableJemalloc = true; };
};
}; };
saiph = { saiph = {