forked from rarias/bscpkgs
hpcg: precompute the input
This commit is contained in:
@@ -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; }
|
||||
|
||||
Reference in New Issue
Block a user