minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Add 5090 dry run stage to PR test workflow (#17022)

Browse Source
This commit is contained in:
Alison Shao
2026-01-13 14:12:33 -08:00
committed by GitHub
parent 339915ce2b
commit b880607108
71 changed files with 158 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/models/test_cross_encoder_models.py
+1
View File
@@ -2,6 +2,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
# Cross encoder model tests
register_cuda_ci(est_time=100, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=105, suite="stage-b-test-small-1-gpu-5090")
register_amd_ci(est_time=150, suite="stage-b-test-small-1-gpu-amd")
import multiprocessing as mp
test/registered/models/test_embedding_models.py
+1
View File
@@ -2,6 +2,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
# Embedding model tests
register_cuda_ci(est_time=73, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=58, suite="stage-b-test-small-1-gpu-5090")
register_amd_ci(
est_time=73,
suite="stage-b-test-small-1-gpu-amd",
test/registered/models/test_encoder_embedding_models.py
+1
View File
@@ -2,6 +2,7 @@ from sglang.test.ci.ci_register import register_cuda_ci
# Encoder embedding model tests (CUDA only)
register_cuda_ci(est_time=270, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=144, suite="stage-b-test-small-1-gpu-5090")
# Copyright 2023-2024 SGLang Team
# Licensed under the Apache License, Version 2.0 (the "License");
test/registered/models/test_qwen_models.py
+1
View File
@@ -2,6 +2,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
# Qwen model tests
register_cuda_ci(est_time=90, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=84, suite="stage-b-test-small-1-gpu-5090")
register_amd_ci(est_time=130, suite="stage-b-test-small-1-gpu-amd")
import unittest
test/registered/models/test_reward_models.py
+1
View File
@@ -2,6 +2,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
# Reward model tests
register_cuda_ci(est_time=103, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=69, suite="stage-b-test-small-1-gpu-5090")
register_amd_ci(est_time=132, suite="stage-b-test-small-1-gpu-amd")
# Copyright 2023-2024 SGLang Team
test/registered/models/test_transformers_models.py
+1
View File
@@ -2,6 +2,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
# Transformers fallback model tests
register_cuda_ci(est_time=245, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=210, suite="stage-b-test-small-1-gpu-5090")
register_amd_ci(est_time=320, suite="stage-b-test-small-1-gpu-amd")
import dataclasses