Skip to content

Commit

Permalink
[TPUs] Update sky show-gpus for TPUs (#3941)
Browse files Browse the repository at this point in the history
* update TPU

* update TPU

* update TPU

* lint
  • Loading branch information
romilbhardwaj authored Sep 13, 2024
1 parent 8f4eef4 commit c1464e1
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions sky/clouds/service_catalog/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -337,10 +337,13 @@ def get_common_gpus() -> List[str]:
def get_tpus() -> List[str]:
"""Returns a list of TPU names."""
# TODO(wei-lin): refactor below hard-coded list.
# There are many TPU configurations available, we show the three smallest
# and the largest configuration for the latest gen TPUs.
return [
'tpu-v2-8', 'tpu-v2-32', 'tpu-v2-128', 'tpu-v2-256', 'tpu-v2-512',
'tpu-v3-8', 'tpu-v3-32', 'tpu-v3-64', 'tpu-v3-128', 'tpu-v3-256',
'tpu-v3-512', 'tpu-v3-1024', 'tpu-v3-2048'
'tpu-v2-512', 'tpu-v3-2048', 'tpu-v4-8', 'tpu-v4-16', 'tpu-v4-32',
'tpu-v4-3968', 'tpu-v5litepod-1', 'tpu-v5litepod-4', 'tpu-v5litepod-8',
'tpu-v5litepod-256', 'tpu-v5p-8', 'tpu-v5p-32', 'tpu-v5p-128',
'tpu-v5p-12288'
]


Expand Down

0 comments on commit c1464e1

Please sign in to comment.