From 7b128e143a5ffcda89fe5759a486f47386b1446e Mon Sep 17 00:00:00 2001 From: "Hsiu-Chun, Hung" <160560375+Emmanuel0612@users.noreply.github.com> Date: Thu, 14 May 2026 15:03:56 +0800 Subject: [PATCH] [AMD] Add amd jit clamp position bench ci (#25209) Co-authored-by: github-actions[bot] Co-authored-by: YC Yen-Ching Tseng --- python/sglang/jit_kernel/benchmark/bench_clamp_position.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/sglang/jit_kernel/benchmark/bench_clamp_position.py b/python/sglang/jit_kernel/benchmark/bench_clamp_position.py index 52082c64b..8b33edc6b 100644 --- a/python/sglang/jit_kernel/benchmark/bench_clamp_position.py +++ b/python/sglang/jit_kernel/benchmark/bench_clamp_position.py @@ -11,9 +11,10 @@ from sglang.jit_kernel.benchmark.utils import ( ) from sglang.jit_kernel.clamp_position import clamp_position_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=13, suite="stage-b-kernel-benchmark-1-gpu-large") +register_amd_ci(est_time=16, suite="jit-kernel-unit-test-amd") SIZE_LIST = get_benchmark_range( full_range=[2**n for n in range(4, 16)],