diff --git a/test/registered/distributed/test_parallel_state.py b/test/registered/distributed/test_parallel_state.py index 3a9209b42..c3610a274 100644 --- a/test/registered/distributed/test_parallel_state.py +++ b/test/registered/distributed/test_parallel_state.py @@ -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") diff --git a/test/registered/input_embedding/test_input_embeds_chunked.py b/test/registered/input_embedding/test_input_embeds_chunked.py index 61c3ca501..f3aad1abf 100644 --- a/test/registered/input_embedding/test_input_embeds_chunked.py +++ b/test/registered/input_embedding/test_input_embeds_chunked.py @@ -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 diff --git a/test/registered/openai_server/basic/test_http2_server.py b/test/registered/openai_server/basic/test_http2_server.py index e2a05a335..3f1434e8b 100644 --- a/test/registered/openai_server/basic/test_http2_server.py +++ b/test/registered/openai_server/basic/test_http2_server.py @@ -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])") diff --git a/test/registered/prefill_only/test_embed_overrides.py b/test/registered/prefill_only/test_embed_overrides.py index f0c8ff830..9c4a0c334 100644 --- a/test/registered/prefill_only/test_embed_overrides.py +++ b/test/registered/prefill_only/test_embed_overrides.py @@ -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 diff --git a/test/registered/radix_cache/test_radix_cache_hit.py b/test/registered/radix_cache/test_radix_cache_hit.py index 18bd3a997..bcd2b2b1c 100644 --- a/test/registered/radix_cache/test_radix_cache_hit.py +++ b/test/registered/radix_cache/test_radix_cache_hit.py @@ -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