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/quant/test_autoround.py
+1
View File
@@ -18,6 +18,7 @@ from sglang.test.test_utils import (
)
register_cuda_ci(est_time=77, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=76, suite="stage-b-test-small-1-gpu-5090")
class TestAutoRound(CustomTestCase):
test/registered/quant/test_block_int8.py
+1
View File
@@ -11,6 +11,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=44, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=25, suite="stage-b-test-small-1-gpu-5090")
register_amd_ci(est_time=22, suite="stage-b-test-small-1-gpu-amd")
test/registered/quant/test_int8_kernel.py
+1
View File
@@ -12,6 +12,7 @@ from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=8, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=10, suite="stage-b-test-small-1-gpu-5090")
def native_w8a8_per_token_matmul(A, B, As, Bs, output_dtype=torch.float16):
test/registered/quant/test_torchao.py
+1
View File
@@ -7,6 +7,7 @@ from sglang import Engine
from sglang.test.ci.ci_register import register_cuda_ci
register_cuda_ci(est_time=103, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=88, suite="stage-b-test-small-1-gpu-5090")
from sglang.lang.chat_template import get_chat_template_by_model_path
from sglang.srt.utils import kill_process_tree
from sglang.test.run_eval import run_eval
test/registered/quant/test_triton_scaled_mm.py
+1
View File
@@ -9,6 +9,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=8, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=9, suite="stage-b-test-small-1-gpu-5090")
register_amd_ci(est_time=12, suite="stage-b-test-small-1-gpu-amd")