Skip to content

Commit 25f31a1

Browse files
authored
fix: remove unsupported gpus from configuration (#1602)
## Description Restricted GPU options in the launcher task annotation schema to only include NVIDIA H200, removing NVIDIA H100 and NVIDIA A100 options. ## Related Issue and Pull requests ## Type of Change - [ ] Bug fix - [ ] New feature - [x] Improvement - [ ] Cleanup/Refactor - [ ] Breaking change - [ ] Documentation update ## Checklist - [ ] I have tested this does not break current pipelines / runs functionality - [ ] I have tested the changes on staging ## Screenshots (if applicable) ## Test Instructions ![image.png](https://app.graphite.com/user-attachments/assets/09c0c80a-5865-4c89-86d9-10228c4e5f42.png) Verify that only NVIDIA H200 appears as a GPU option in the launcher task interface. ## Additional Comments This change simplifies GPU selection by limiting options to only the H200 model. See https://vault.shopify.io/docs/craft/138-Data/data_handbook/ml_at_shopify/ml_infrastructure#shopify-gpu-policy
1 parent fd2bdad commit 25f31a1

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/config/launcherTaskAnnotationSchema.json

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,11 +75,9 @@
7575
"description": "GPU type and quantity",
7676
"x-enable-quantity": true,
7777
"exclusiveMinimum": 0,
78-
"enum": ["NVIDIA-H200", "NVIDIA-H100", "NVIDIA-A100"],
78+
"enum": ["NVIDIA-H200"],
7979
"x-enum-labels": {
80-
"NVIDIA-H200": "NVIDIA H200",
81-
"NVIDIA-H100": "NVIDIA H100",
82-
"NVIDIA-A100": "NVIDIA A100"
80+
"NVIDIA-H200": "NVIDIA H200"
8381
}
8482
}
8583
}

0 commit comments

Comments
 (0)