Skip to content

Commit

Permalink
add default gpu-partition-size label to allow scheduling on multi-ins…
Browse files Browse the repository at this point in the history
…tance GPUs in GKE

Signed-off-by: Cornelis Boon <[email protected]>
  • Loading branch information
cjidboon94 committed Mar 21, 2024
1 parent 8a6d202 commit 83fac0c
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions charts/flyte-binary/gke-starter.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down

0 comments on commit 83fac0c

Please sign in to comment.