hpcg: precompute the input

This commit is contained in:
2020-11-09 17:47:55 +01:00
parent 5763b91d39
commit 966606b62d
3 changed files with 119 additions and 4 deletions

View File

@@ -4,6 +4,7 @@
, bsc
, targetMachine
, stages
, genInput
}:
with stdenv.lib;
@@ -13,8 +14,8 @@ let
varConf = with bsc; {
# FIXME: Temporally reduce the input size until we can load a precomputed
# input in each run, otherwise the execution time is very large.
n = [ { x = 104; y = 104; z = 104; } ];
#n = [ { x = 256; y = 288; z = 288; } ];
#n = [ { x = 104; y = 104; z = 104; } ];
n = [ { x = 256; y = 288; z = 288; } ];
nblocks = [ 12 24 48 96 192 384 ];
};
@@ -50,6 +51,8 @@ let
inherit varConf genConf;
};
input = genInput configs;
exec = {nextStage, conf, ...}: with conf; stages.exec {
inherit nextStage;
env = "NANOS6_DEPENDENCIES=discrete";
@@ -58,6 +61,9 @@ let
"--ny=${toString n.y}"
"--nz=${toString n.z}"
"--nblocks=${toString nblocks}"
# The input symlink is generated by the input stage, which is generated by
# the genInput function.
"--load=input"
];
};
@@ -69,8 +75,9 @@ let
inherit cc nanos6 mcxx gitBranch;
};
pipeline = stdexp.stdPipeline ++ [ exec program ];
pipeline = stdexp.stdPipeline ++ [ input exec program ];
in
#{ inherit configs pipeline; }
stdexp.genExperiment { inherit configs pipeline; }