Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

[k8s] Fix GKELabelFormatter for H100s #3627

Merged
merged 2 commits into from
Jun 4, 2024
Merged

[k8s] Fix GKELabelFormatter for H100s #3627

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8a16ea2
H100-80gb does not exist, fix to H100
romilbhardwaj Jun 3, 2024
f026756
Fix H100 support
romilbhardwaj Jun 3, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions sky/provision/kubernetes/utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -98,10 +98,10 @@ def get_gke_accelerator_name(accelerator: str) -> str:
"""Returns the accelerator name for GKE clusters

Uses the format - nvidia-tesla-<accelerator>.
A100-80GB, H100-80GB and L4 are an exception. They use nvidia-<accelerator>.
A100-80GB, H100 and L4 are an exception. They use nvidia-<accelerator>.
"""
if accelerator in ('A100-80GB', 'L4', 'H100-80GB'):
# A100-80GB, L4 and H100-80GB have a different name pattern.
if accelerator in ('A100-80GB', 'L4', 'H100'):
# A100-80GB, L4 and H100 have a different name pattern.
return 'nvidia-{}'.format(accelerator.lower())
else:
return 'nvidia-tesla-{}'.format(accelerator.lower())
Expand Down
Loading