From 171ab6d1c521e602d0e611f8604c83b89a3769fe Mon Sep 17 00:00:00 2001 From: Cornelis Boon Date: Wed, 20 Mar 2024 12:28:29 +0100 Subject: [PATCH] add default gpu-partition-size label to allow scheduling on multi-instance GPUs in GKE Signed-off-by: Cornelis Boon --- charts/flyte-binary/gke-starter.yaml | 1 + 1 file changed, 1 insertion(+) diff --git a/charts/flyte-binary/gke-starter.yaml b/charts/flyte-binary/gke-starter.yaml index aabc4426a1e..f9052a27fd7 100644 --- a/charts/flyte-binary/gke-starter.yaml +++ b/charts/flyte-binary/gke-starter.yaml @@ -80,6 +80,7 @@ configuration: k8s: inject-finalizer: true gpu-device-node-label: cloud.google.com/gke-accelerator + gpu-partition-size-node-label: cloud.google.com/gke-gpu-partition-size resource-tolerations: - nvidia.com/gpu: - key: "nvidia.com/gpu"