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

[AMD] Add amd jit clamp position bench ci (#25209)

Browse Source 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: YC Yen-Ching Tseng <yctseng@amd.com>
This commit is contained in:
Hsiu-Chun, Hung
2026-05-14 00:03:56 -07:00
committed by GitHub
co-authored by github-actions[bot] YC Yen-Ching Tseng
parent 41eb2d861b
commit 7b128e143a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/jit_kernel/benchmark/bench_clamp_position.py
+2 -1
View File
@@ -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)],