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

[AMD] batch 3: register newly-added JIT kernel benchmarks for jit-kernel-benchmark-test-amd (#31615)

Browse Source
This commit is contained in:
Michael
2026-07-20 23:51:51 -07:00
committed by GitHub
parent d6ef68881e
commit c4c405a46b
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/jit/benchmark/bench_spec_topk1.py
+2 -1
View File
@@ -12,11 +12,12 @@ from sglang.jit_kernel.benchmark.utils import (
run_benchmark, run_benchmark,
) )
from sglang.kernels.ops.speculative.topk1 import draft_topk1_postprocess from sglang.kernels.ops.speculative.topk1 import draft_topk1_postprocess
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( register_cuda_ci(
est_time=30, stage="base-b-kernel-benchmark", runner_config="1-gpu-large" est_time=30, stage="base-b-kernel-benchmark", runner_config="1-gpu-large"
) )
register_amd_ci(est_time=30, stage="jit-kernel-benchmark", runner_config="amd")
BATCH_SIZE_RANGE = get_benchmark_range( BATCH_SIZE_RANGE = get_benchmark_range(
test/registered/jit/benchmark/bench_vocab_parallel_embedding.py
+2 -1
View File
@@ -8,11 +8,12 @@ from sglang.kernels.ops.embeddings.vocab_parallel_embedding import (
vocab_parallel_embedding, vocab_parallel_embedding,
) )
from sglang.srt.layers.vocab_parallel_embedding import get_masked_input_and_mask from sglang.srt.layers.vocab_parallel_embedding import get_masked_input_and_mask
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( register_cuda_ci(
est_time=10, stage="base-b-kernel-benchmark", runner_config="1-gpu-large" est_time=10, stage="base-b-kernel-benchmark", runner_config="1-gpu-large"
) )
register_amd_ci(est_time=10, stage="jit-kernel-benchmark", runner_config="amd")
# Key order must match the perf_report x_names. # Key order must match the perf_report x_names.
DEFAULTS = dict( DEFAULTS = dict(