pattern match against A100'

skypilot-org · May 2, 2024 · 3da382c · 3da382c
1 parent db9a091
commit 3da382c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sky/provision/kubernetes/utils.py b/sky/provision/kubernetes/utils.py
@@ -176,7 +176,10 @@ def get_accelerator_from_label_value(cls, value: str) -> str:
  ]
 
  for canonical_name in canonical_gpu_names:
- if canonical_name in value:
+ # A100-80G accelerator can be labeled as A100-SXM-80GB or A100-PCIE-80GB
+ if canonical_name == 'A100-80G' and if re.match(r'A100.*-80G', value):
+ return canonical_name
+ elif canonical_name in value:
  return canonical_name
 
  # If we didn't find a canonical name: