2020-11-09 17:47:55 +01:00
|
|
|
{
|
|
|
|
stdenv
|
|
|
|
, stdexp
|
|
|
|
, bsc
|
|
|
|
, targetMachine
|
|
|
|
, stages
|
|
|
|
, garlicTools
|
2021-04-16 11:51:34 +02:00
|
|
|
, callPackage
|
2020-11-09 17:47:55 +01:00
|
|
|
}:
|
|
|
|
|
|
|
|
with stdenv.lib;
|
|
|
|
with garlicTools;
|
|
|
|
|
2021-04-16 11:51:34 +02:00
|
|
|
rec {
|
|
|
|
|
2020-11-09 17:47:55 +01:00
|
|
|
# Generate the complete configuration for each unit
|
2021-02-23 18:24:21 +01:00
|
|
|
genConf = c: targetMachine.config // rec {
|
2021-04-16 16:15:16 +02:00
|
|
|
expName = "hpcg-gen";
|
|
|
|
unitName = expName
|
|
|
|
+ "-nodes${toString nodes}"
|
|
|
|
+ "-spt.z${toString sizePerTask.z}";
|
2020-11-09 17:47:55 +01:00
|
|
|
|
2020-11-17 11:17:57 +01:00
|
|
|
inherit (targetMachine.config) hw;
|
2020-11-09 17:47:55 +01:00
|
|
|
|
2021-04-16 11:51:34 +02:00
|
|
|
# Inherit options from the current conf
|
|
|
|
inherit (c) sizePerTask nprocs disableAspectRatio gitBranch
|
|
|
|
cpusPerTask ntasksPerNode nodes;
|
|
|
|
|
|
|
|
# nblocks and ncomms are ignored from c
|
|
|
|
ncomms = 1;
|
|
|
|
nblocks = 1;
|
2020-11-09 17:47:55 +01:00
|
|
|
|
2021-04-16 11:51:34 +02:00
|
|
|
# We only need one run
|
2020-11-09 17:47:55 +01:00
|
|
|
loops = 1;
|
|
|
|
|
2021-04-16 11:51:34 +02:00
|
|
|
# Generate the input
|
|
|
|
enableGen = true;
|
|
|
|
|
2020-11-09 17:47:55 +01:00
|
|
|
# Resources
|
|
|
|
qos = "debug";
|
2021-04-16 11:51:34 +02:00
|
|
|
time = "02:00:00";
|
2020-11-09 17:47:55 +01:00
|
|
|
jobName = unitName;
|
|
|
|
};
|
|
|
|
|
2021-04-16 11:51:34 +02:00
|
|
|
common = callPackage ./common.nix {};
|
2020-11-09 17:47:55 +01:00
|
|
|
|
2021-04-16 11:51:34 +02:00
|
|
|
getInputTre = conf: stdexp.genExperiment {
|
|
|
|
configs = [ (genConf conf) ];
|
|
|
|
pipeline = common.pipeline;
|
2021-02-23 18:24:21 +01:00
|
|
|
};
|
2021-04-16 11:51:34 +02:00
|
|
|
}
|