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

[AMD] Fix CI - unit-test-backend-1-gpu-amd-mi35x and unit-test-backend-2-gpu-amd, stage-b-test-small-1-gpu-amd (#16675)

Browse Source 
Co-authored-by: Bingxu Chen <bingxche@amd.com>
This commit is contained in:
YC Tseng
2026-01-08 10:16:27 -08:00
committed by GitHub
co-authored by Bingxu Chen
parent fbc24886dd
commit 4e999404c3
60 changed files with 106 additions and 143 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/models/test_compressed_tensors_models.py
+1 -1
View File
@@ -2,7 +2,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
# Model tests for compressed tensors (FP8)
register_cuda_ci(est_time=42, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=42, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=42, suite="stage-b-test-small-1-gpu-amd")
import unittest
from types import SimpleNamespace
test/registered/models/test_cross_encoder_models.py
+1 -1
View File
@@ -2,7 +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_amd_ci(est_time=150, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=150, suite="stage-b-test-small-1-gpu-amd")
import multiprocessing as mp
import random
test/registered/models/test_embedding_models.py
+1 -1
View File
@@ -4,7 +4,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=73, suite="stage-b-test-small-1-gpu")
register_amd_ci(
est_time=73,
suite="stage-b-test-small-1-gpu",
suite="stage-b-test-small-1-gpu-amd",
disabled="see https://github.com/sgl-project/sglang/issues/11127",
)
test/registered/models/test_qwen_models.py
+1 -1
View File
@@ -2,7 +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_amd_ci(est_time=130, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=130, suite="stage-b-test-small-1-gpu-amd")
import unittest
from types import SimpleNamespace
test/registered/models/test_reward_models.py
+1 -1
View File
@@ -2,7 +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_amd_ci(est_time=132, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=132, suite="stage-b-test-small-1-gpu-amd")
# Copyright 2023-2024 SGLang Team
# Licensed under the Apache License, Version 2.0 (the "License");
test/registered/models/test_transformers_models.py
+1 -1
View File
@@ -2,7 +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_amd_ci(est_time=320, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=320, suite="stage-b-test-small-1-gpu-amd")
import dataclasses
import multiprocessing as mp
test/registered/models/test_vlm_models.py
+1 -1
View File
@@ -2,7 +2,7 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
# VLM (Vision Language Model) tests
register_cuda_ci(est_time=270, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=420, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=420, suite="stage-b-test-small-1-gpu-amd")
import argparse
import random