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

[AMD] Register 5 server-style 1-GPU tests for AMD PR CI (#24572)

Browse Source
This commit is contained in:
Michael
2026-05-12 22:45:52 -07:00
committed by GitHub
parent c665edec6e
commit 245f7d8026
5 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/distributed/test_parallel_state.py
+2 -1
View File
@@ -41,9 +41,10 @@ from unittest.mock import Mock, patch
import pytest
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=8, suite="stage-b-test-1-gpu-small")
register_amd_ci(est_time=8, suite="stage-b-test-1-gpu-small-amd")
# Import the actual parallel_state module
parallel_state = pytest.importorskip("sglang.srt.distributed.parallel_state")
test/registered/input_embedding/test_input_embeds_chunked.py
+2 -1
View File
@@ -21,7 +21,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
from sglang.srt.environ import envs
from sglang.srt.utils import kill_process_tree
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.test_utils import (
DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
@@ -31,6 +31,7 @@ from sglang.test.test_utils import (
)
register_cuda_ci(est_time=43, suite="stage-b-test-1-gpu-small")
register_amd_ci(est_time=43, suite="stage-b-test-1-gpu-small-amd")
CHUNKED_PREFILL_SIZE = 256
test/registered/openai_server/basic/test_http2_server.py
+2 -1
View File
@@ -11,7 +11,7 @@ import unittest
import requests
from sglang.srt.utils import kill_process_tree
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.test_utils import (
DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
@@ -28,6 +28,7 @@ except ImportError:
_HAS_GRANIAN = False
register_cuda_ci(est_time=52, suite="stage-b-test-1-gpu-small")
register_amd_ci(est_time=52, suite="stage-b-test-1-gpu-small-amd")
@unittest.skipUnless(_HAS_GRANIAN, "granian not installed (pip install sglang[http2])")
test/registered/prefill_only/test_embed_overrides.py
+2 -1
View File
@@ -21,10 +21,11 @@ from sglang.srt.managers.tokenizer_manager_score_mixin import (
TokenizerManagerScoreMixin,
)
from sglang.srt.server_args import MIS_DELIMITER_TOKEN_ID
from sglang.test.ci.ci_register import register_cuda_ci
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=9, suite="stage-b-test-1-gpu-small")
register_amd_ci(est_time=9, suite="stage-b-test-1-gpu-small-amd")
HIDDEN_DIM = 4
test/registered/radix_cache/test_radix_cache_hit.py
+2 -1
View File
@@ -1,7 +1,7 @@
import unittest
from sglang.srt.utils import kill_process_tree
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.kits.cache_hit_kit import run_multiturn_cache_hit_test
from sglang.test.test_utils import (
DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
@@ -12,6 +12,7 @@ from sglang.test.test_utils import (
)
register_cuda_ci(est_time=55, suite="stage-b-test-1-gpu-small")
register_amd_ci(est_time=55, suite="stage-b-test-1-gpu-small-amd")
MODEL = DEFAULT_SMALL_MODEL_NAME_FOR_TEST