Add Slurm task prolog/epilog patch to image build

abbas1902 · abbas1902 · commit 5ea44de3e290 · 2025-06-23T16:57:33.000Z
diff --git a/examples/machine-learning/build-service-images/a3m/blueprint.yaml b/examples/machine-learning/build-service-images/a3m/blueprint.yaml
@@ -57,6 +57,7 @@ deployment_groups:
             "install_managed_lustre": true,
             "install_gcsfuse": true,
             "monitoring_agent": "cloud-ops",
+            "slurm_patch_files": ["task_prolog_epilog.patch"],
             "use_open_drivers": true
           }
       - type: shell
diff --git a/examples/machine-learning/build-service-images/common/blueprint.yaml b/examples/machine-learning/build-service-images/common/blueprint.yaml
@@ -48,6 +48,7 @@ deployment_groups:
             "install_managed_lustre": true,
             "install_nvidia_repo": true,
             "install_ompi": true,
+            "slurm_patch_files": ["task_prolog_epilog.patch"],
             "allow_kernel_upgrades": false,
             "monitoring_agent": "cloud-ops",
           }
diff --git a/examples/machine-learning/build-service-images/shared.yaml b/examples/machine-learning/build-service-images/shared.yaml
@@ -16,7 +16,7 @@ vars:
   region: us-central1
   zone: us-central1-a
 
-  slurm_gcp_version: 6.10.0
+  slurm_gcp_version: 6.10.4
 
   # to supply
   family:

Original file line number	Diff line number	Diff line change
`@@ -57,6 +57,7 @@ deployment_groups:`
`57`	`57`	`"install_managed_lustre": true,`
`58`	`58`	`"install_gcsfuse": true,`
`59`	`59`	`"monitoring_agent": "cloud-ops",`
	`60`	`+ "slurm_patch_files": ["task_prolog_epilog.patch"],`
`60`	`61`	`"use_open_drivers": true`
`61`	`62`	`}`
`62`	`63`	`- type: shell`
Original file line number	Diff line number	Diff line change
`@@ -48,6 +48,7 @@ deployment_groups:`
`48`	`48`	`"install_managed_lustre": true,`
`49`	`49`	`"install_nvidia_repo": true,`
`50`	`50`	`"install_ompi": true,`
	`51`	`+ "slurm_patch_files": ["task_prolog_epilog.patch"],`
`51`	`52`	`"allow_kernel_upgrades": false,`
`52`	`53`	`"monitoring_agent": "cloud-ops",`
`53`	`54`	`}`