Slurm prolog only in benchmark mode

Author: heerener

hpc_provisioner/src/hpc_provisioner/config-dev/_head_node.tpl.yaml

@@ -21,7 +21,6 @@ CustomActions:
         Args: !config create_users_args
       - Script: s3://sboinfrastructureassets-sandbox/scripts/environment.sh
         Args: !config environment_args
-      - Script: s3://sboinfrastructureassets-sandbox/scripts/80_cloudwatch_agent_config_prolog.sh
 #       - Script: s3://sboinfrastructureassets/scripts/all_or_nothing_allocation.sh
 #       - Script: s3://sboinfrastructureassets/scripts/setup_lustre_directories.py
 #         Args:

hpc_provisioner/src/hpc_provisioner/pcluster_manager.py

@@ -208,6 +208,13 @@ def pcluster_create(
     pcluster_config["Scheduling"]["SlurmQueues"] = choose_tier(pcluster_config, options)
     if not options["include_lustre"]:
         pcluster_config["SharedStorage"].pop(1)
+    if benchmark:
+        pcluster_config["HeadNode"]["CustomActions"]["OnNodeConfigured"]["Sequence"].append(
+            {
+                "Script": "s3://sboinfrastructureassets-sandbox/scripts/80_cloudwatch_agent_config_prolog.sh"
+            }
+        )
+
     output_file_name = write_config(cluster_name, pcluster_config)
 
     try: