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

[CI] Move misplaced mhc kernel test into test/registered/kernels (#27781)

Browse Source 
Co-authored-by: Claude <noreply@anthropic.com>
This commit is contained in:
DarkSharpness
2026-06-10 01:49:18 -07:00
committed by GitHub
co-authored by Claude
parent 70c71ba183
commit b40f365732
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/kernels/test_mhc_kernels.py → test/registered/kernels/test_mhc_kernels.py
+9
View File
@@ -3,6 +3,9 @@ import torch
import sglang.srt.layers.mhc as mhc import sglang.srt.layers.mhc as mhc
from sglang.srt.layers.mhc import mhc_fused_post_pre, mhc_post, mhc_pre from sglang.srt.layers.mhc import mhc_fused_post_pre, mhc_post, mhc_pre
from sglang.test.ci.ci_register import register_cuda_ci
register_cuda_ci(est_time=30, stage="base-b", runner_config="1-gpu-large")
@pytest.mark.parametrize("hidden_size", [4096, 7168]) @pytest.mark.parametrize("hidden_size", [4096, 7168])
@@ -109,3 +112,9 @@ def test_mhc_fused_post_pre_matches_unfused(
layer_atol = 2e-2 if use_norm else 2e-3 layer_atol = 2e-2 if use_norm else 2e-3
layer_rtol = 2e-2 if use_norm else 2e-3 layer_rtol = 2e-2 if use_norm else 2e-3
torch.testing.assert_close(layer_out, layer_ref, atol=layer_atol, rtol=layer_rtol) torch.testing.assert_close(layer_out, layer_ref, atol=layer_atol, rtol=layer_rtol)
if __name__ == "__main__":
import sys
sys.exit(pytest.main([__file__]))