feat: use sgl-kernel cu129 as default (#10188)
This commit is contained in:
@@ -138,9 +138,13 @@ def test_int4_fp8_grouped_gemm_single_expert(batch_size):
|
||||
raise
|
||||
|
||||
|
||||
# @pytest.mark.skipif(
|
||||
# not is_hopper(),
|
||||
# reason="cutlass_w4a8_moe_mm is only supported on sm90",
|
||||
# )
|
||||
@pytest.mark.skipif(
|
||||
not is_hopper(),
|
||||
reason="cutlass_w4a8_moe_mm is only supported on sm90",
|
||||
True,
|
||||
reason="TODO(rainj-me): fix cu129 binary issue on hopper cu126",
|
||||
)
|
||||
@pytest.mark.parametrize("batch_size", [2, 4, 8, 16])
|
||||
@pytest.mark.parametrize("k", [256, 512, 1024])
|
||||
|
||||
Reference in New Issue
Block a user