Skip to content

Commit 5e45f4a

Browse files
authored
Fix helm chart nodeSelector for GPU endpoints (#430)
1 parent b0e7d29 commit 5e45f4a

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

charts/model-engine/templates/service_template_config_map.yaml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,9 @@ data:
9292
{{- toYaml . | nindent 12 }}
9393
{{- end }}
9494
{{- if eq $device "gpu" }}
95+
{{- if empty $node_selector }}
96+
nodeSelector:
97+
{{- end }}
9598
k8s.amazonaws.com/accelerator: ${GPU_TYPE}
9699
tolerations:
97100
- key: "nvidia.com/gpu"
@@ -634,6 +637,9 @@ data:
634637
{{- toYaml . | nindent 12 }}
635638
{{- end }}
636639
{{- if eq $device "gpu" }}
640+
{{- if empty $node_selector }}
641+
nodeSelector:
642+
{{- end }}
637643
k8s.amazonaws.com/accelerator: ${GPU_TYPE}
638644
tolerations:
639645
- key: "nvidia.com/gpu"

0 commit comments

Comments
 (0)