Skip to content

Commit

Permalink
pattern match against A100'
Browse files Browse the repository at this point in the history
  • Loading branch information
asaiacai committed May 2, 2024
1 parent 3da382c commit 4be3589
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sky/provision/kubernetes/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -177,7 +177,7 @@ def get_accelerator_from_label_value(cls, value: str) -> str:

for canonical_name in canonical_gpu_names:
# 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):
if canonical_name == 'A100-80G' and re.match(r'A100.*-80G', value):
return canonical_name
elif canonical_name in value:
return canonical_name
Expand Down

0 comments on commit 4be3589

Please sign in to comment.