Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
asaiacai committed Aug 29, 2024
1 parent a3fdb77 commit 922ac24
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion sky/provision/kubernetes/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,8 @@ def get_gke_accelerator_name(accelerator: str) -> str:
# H100 is named as H100-80GB in GKE.
accelerator = 'H100-80GB'
if accelerator in ('A100-80GB', 'L4', 'H100-80GB', 'H100-MEGA-80GB'):
# A100-80GB, L4, H100-80GB and H100-MEGA-80GB have a different name pattern.
# A100-80GB, L4, H100-80GB and H100-MEGA-80GB
# have a different name pattern.
return 'nvidia-{}'.format(accelerator.lower())
else:
return 'nvidia-tesla-{}'.format(accelerator.lower())
Expand Down

0 comments on commit 922ac24

Please sign in to comment.