Skip to content

Commit e40036b

Browse files
Use short GPU names in scripts (#245)
Server-side GPU identifiers are moving from "NVIDIA TESLA T4", "NVIDIA H100", … to short forms ("T4", "H100", …). Update the deploy/run scripts to send the new names. See AlphaGPU/leetgpu-infra#343. Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent ca891d3 commit e40036b

2 files changed

Lines changed: 2 additions & 4 deletions

File tree

scripts/run_challenge.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,7 @@ def main() -> int:
9595
parser = argparse.ArgumentParser(description="Submit a solution via WebSocket API.")
9696
parser.add_argument("challenge_path", type=Path, help="Path to the challenge directory")
9797
parser.add_argument("--language", default="cuda", help="Language (default: cuda)")
98-
parser.add_argument(
99-
"--gpu", default="NVIDIA TESLA T4", help="GPU name (default: NVIDIA TESLA T4)"
100-
)
98+
parser.add_argument("--gpu", default="T4", help="GPU name (default: T4)")
10199
parser.add_argument(
102100
"--action", default="run", choices=["run", "submit"], help="Action (run or submit)"
103101
)

scripts/update_challenges.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
SERVICE_URL = os.getenv("SERVICE_URL", "http://localhost:8080")
2626
LEETGPU_API_KEY = os.getenv("LEETGPU_API_KEY")
2727

28-
GPUS = ["NVIDIA H100", "NVIDIA H200", "NVIDIA TESLA T4", "NVIDIA B200", "NVIDIA A100-80GB"]
28+
GPUS = ["H100", "H200", "T4", "B200", "A100-80GB"]
2929

3030

3131
def extract_id(name: str) -> int:

0 commit comments

Comments
 (0)