From bc265c5f82fc4a7612775e4d7a934b7c947d539b Mon Sep 17 00:00:00 2001 From: "Hsiu-Chun, Hung" <160560375+Emmanuel0612@users.noreply.github.com> Date: Thu, 14 May 2026 15:13:12 +0800 Subject: [PATCH] [AMD] Add amd jit resolve token ids bench ci (#25210) Co-authored-by: github-actions[bot] Co-authored-by: YC Yen-Ching Tseng --- .../jit_kernel/benchmark/bench_resolve_future_token_ids.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py b/python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py index f56c8df2a..5551d2968 100644 --- a/python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py +++ b/python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py @@ -11,9 +11,10 @@ from sglang.jit_kernel.benchmark.utils import ( ) from sglang.jit_kernel.resolve_future_token_ids import resolve_future_token_ids_cuda from sglang.srt.utils import get_compiler_backend -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=10, suite="stage-b-kernel-benchmark-1-gpu-large") +register_amd_ci(est_time=10, suite="jit-kernel-unit-test-amd") SIZE_LIST = get_benchmark_range( full_range=[2**n for n in range(4, 16)], # 16 … 32K elements