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

[AMD] Fix jit-kernel-unit-test-amd: activation.cuh ROCm build + per_token CUDA-only (R165) (#27947)

Browse Source
This commit is contained in:
Michael
2026-06-15 23:44:26 -07:00
committed by GitHub
parent 556cf54d47
commit 72d962be88
3 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/jit/test_per_token_group_quant_8bit.py
+1 -2
View File
@@ -23,11 +23,10 @@ from sglang.srt.layers.quantization.fp8_kernel import (
from sglang.srt.layers.quantization.fp8_kernel import (
per_token_group_quant_8bit as triton_per_token_group_quant_8bit,
)
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.ci.ci_register import register_cuda_ci
register_cuda_ci(est_time=16, suite="base-b-kernel-unit-1-gpu-large")
register_cuda_ci(est_time=120, suite="nightly-kernel-1-gpu", nightly=True)
register_amd_ci(est_time=16, suite="jit-kernel-unit-test-amd")
configs = list(
itertools.product(