bscpkgs/garlic/exp/hpcg/scaling.nix

{
  stdenv
, stdexp
, bsc
, targetMachine
, stages
, garlicTools
, callPackage
, enableExtended ? false
, enableStrong ? true
}:

with stdenv.lib;
with garlicTools;

let
  common = callPackage ./common.nix { };

  inherit (common) pipeline getSizePerTask;

  maxNodes = 16;

  # Initial variable configuration
  varConf = {
    nodes = range2 1 maxNodes;
    baseSizeZ = if (enableExtended) then [ 8 16 ] else [ 16 ];
    blocksPerCpu = if (enableExtended) then range2 1 8 else [ 4 ];
    gitBranch = [
      "garlic/tampi+isend+oss+task"
    ];
  };

  # Generate the complete configuration for each unit
  genConf = c: targetMachine.config // rec {
    expName = if (enableStrong) then "hpcg-ss" else "hpcg-ws";
    unitName = "${expName}"
    + "-nodes${toString nodes}"
    + "-bpc${toString blocksPerCpu}";

    inherit (targetMachine.config) hw;

    inherit maxNodes;
    sizeFactor = if (enableStrong) then maxNodes / nodes else 1;
    
    # hpcg options
    inherit (c) nodes blocksPerCpu gitBranch;
    totalTasks = ntasksPerNode * nodes;
    sizePerCpu = {
      x = 2;
      y = 2;
      z = c.baseSizeZ * sizeFactor;
    };
    sizePerTask = getSizePerTask cpusPerTask sizePerCpu;
    nprocs = { x=1; y=1; z=totalTasks; };
    nblocks = blocksPerCpu * cpusPerTask;
    ncomms = 1;
    disableAspectRatio = true;

    # Repeat the execution of each unit several times
    loops = 10;

    # Resources
    qos = "debug";
    time = "02:00:00";
    cpusPerTask = hw.cpusPerSocket;
    ntasksPerNode = hw.socketsPerNode;
    jobName = unitName;
  };

  # Compute the array of configurations
  configs = stdexp.buildConfigs {
    inherit varConf genConf;
  };

in

  stdexp.genExperiment { inherit configs pipeline; }