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

Fix Stage B CUDA CI (#28367)

Browse Source 
Co-authored-by: Brayden Zhong <brayden@radixark.ai>
This commit is contained in:
Brayden Zhong
2026-06-16 02:39:17 -07:00
committed by GitHub
co-authored by Brayden Zhong
parent a362ba9da3
commit 25e696aa8d
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/kernels/test_mhc_kernels.py
+1
View File
@@ -18,6 +18,7 @@ def test_mhc_fused_post_pre_matches_unfused(
pytest.skip("CUDA is required for TileLang mHC kernels") pytest.skip("CUDA is required for TileLang mHC kernels")
monkeypatch.setattr(mhc, "is_dsa_prefill_cp_round_robin_split", lambda: False) monkeypatch.setattr(mhc, "is_dsa_prefill_cp_round_robin_split", lambda: False)
monkeypatch.setattr(mhc, "_mhc_pre_warmed", True)
torch.manual_seed(0) torch.manual_seed(0)
device = torch.device("cuda") device = torch.device("cuda")
hc_mult = 4 hc_mult = 4