[AMD][CI] Register Eagle constrained decoding test (#25260)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: YC Yen-Ching Tseng <yctseng@amd.com> Co-authored-by: HAI <hixiao@gmail.com>
This commit is contained in:
co-authored by
github-actions[bot]
YC Yen-Ching Tseng
HAI
parent
4ef9bad223
commit
52875ab6f4
@@ -2,7 +2,7 @@ import unittest
|
|||||||
|
|
||||||
from sglang.srt.environ import envs
|
from sglang.srt.environ import envs
|
||||||
from sglang.srt.utils import kill_process_tree
|
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.json_constrained_kit import JSONConstrainedMixin
|
from sglang.test.kits.json_constrained_kit import JSONConstrainedMixin
|
||||||
from sglang.test.kits.regex_constrained_kit import RegexConstrainedMixin
|
from sglang.test.kits.regex_constrained_kit import RegexConstrainedMixin
|
||||||
from sglang.test.test_utils import (
|
from sglang.test.test_utils import (
|
||||||
@@ -15,6 +15,7 @@ from sglang.test.test_utils import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
register_cuda_ci(est_time=116, stage="base-b", runner_config="1-gpu-large")
|
register_cuda_ci(est_time=116, stage="base-b", runner_config="1-gpu-large")
|
||||||
|
register_amd_ci(est_time=165, stage="stage-b", runner_config="1-gpu-large-amd")
|
||||||
|
|
||||||
|
|
||||||
class TestEagleConstrainedDecoding(
|
class TestEagleConstrainedDecoding(
|
||||||
|
|||||||
Reference in New Issue
Block a user