Enable train size to SPEC CPU benchmarks.

pkgs/spec-cpu/default.nix

@@ -10,7 +10,7 @@
 , autoPatchelfHook
 , libxcrypt-legacy
 , lib
-, benchSize ? "test"
+, benchSize ? "train"
 }:
 
 stdenv.mkDerivation rec {

pkgs/spec-cpu/launcher-vm.sh

@@ -0,0 +1,71 @@
+#!/bin/bash
+
+set -e
+
+if [ -z "$SPEC" ]; then
+  SPEC=$(spec-cpu-mini)
+fi
+
+if [ -z "$SPEC" ]; then
+  echo "cannot find spec, set SPEC variable"
+  exit 1
+fi
+
+where=$TMPDIR
+if [ -z "$where" ]; then
+  if [ -d /tmp ]; then
+    where=/tmp
+  else
+    where=$PWD
+  fi
+fi
+
+cwd=$(readlink -f $where)
+# Place the outcome here
+wd="$cwd/spec"
+mkdir -p "$wd"
+
+benchniter=1
+benchsize=train
+benchtune=base
+
+echo "--- Placing output in $wd ---"
+
+printf 'benchmark\tsize\ttune\titer\ttime_s\n' > "$wd/time.csv"
+
+for srcbench in $SPEC/benchspec/CPU/*; do
+  name=$(basename $srcbench)
+  bench="$wd/$name"
+  bench_out="$wd/$name.csv"
+  rm -rf "$bench"
+  cp -r "$srcbench" "$bench"
+  chmod +w -R "$bench"
+
+  rundir="$bench/run/run_${benchtune}_${benchsize}_nix-m64.0000"
+  echo $rundir
+  sed -i '/^-C/d' "$rundir/speccmds.cmd"
+  echo "--- Running $name for $benchniter iterations ---"
+  (
+    #set -x
+    #cd $rundir
+    ./specinvoke -i $benchniter -E speccmds.cmd > /dev/null
+    #set +x
+  )
+  # Print time
+  awk '/^run [0-9]* elapsed time/{printf \
+    "%s\t%s\t%s\t%s\t%s\n", \
+    "'$name'","'$benchsize'","'$benchtune'",$2,$7}' \
+    "$rundir/speccmds.out" > "$bench_out"
+
+  cat "$bench_out"
+
+  # Accumulate in main CSV
+  cat "$bench_out" >> "$wd/time.csv"
+
+  # Erase intermediate files as they occupy RAM.
+  rm -rf "$bench"
+done
+
+echo "--- RESULTS in $wd/time.csv ---"
+cat "$wd/time.csv"
+echo "---------------------------------------"

pkgs/spec-cpu/launcher.sh

@@ -26,7 +26,7 @@ wd="$cwd/spec"
 mkdir -p "$wd"
 
 benchniter=1
-benchsize=test
+benchsize=train
 benchtune=base
 
 echo "--- Placing output in $wd ---"
@@ -42,6 +42,7 @@ for srcbench in $SPEC/benchspec/CPU/*; do
   chmod +w -R "$bench"
 
   rundir="$bench/run/run_${benchtune}_${benchsize}_nix-m64.0000"
+  echo $rundir
   sed -i '/^-C/d' "$rundir/speccmds.cmd"
   echo "--- Running $name for $benchniter iterations ---"
   (

pkgs/spec-cpu/speclaunch-vm.nix

@@ -0,0 +1,21 @@
+{
+  stdenv
+, bash
+}:
+
+stdenv.mkDerivation {
+  name = "speclaunch-vm";
+  src = ./launcher-vm.sh;
+  dontUnpack = true;
+  dontConfigure = true;
+  dontBuild = true;
+  installPhase = ''
+    mkdir -p $out/bin
+    cp $src $out/bin/speclaunch
+    chmod +x $out/bin/speclaunch
+  '';
+  buildInputs = [ bash ];
+  enableParallelBuilding = false;
+  hardeningDisable = [ "all" ];
+  dontStrip = true;
+}