[JIT kernel] CI support jit kernel tests (#15939)

2025-12-28 23:11:02 +08:00
parent e6d5a213ad
commit 7d02c8e59f
2 changed files with 32 additions and 1 deletions
--- a/python/sglang/jit_kernel/tests/test_per_tensor_quant_fp8.py
+++ b/python/sglang/jit_kernel/tests/test_per_tensor_quant_fp8.py
@@ -43,7 +43,7 @@ def torch_scaled_fp8_quant(tensor, inv_scale):
    "num_tokens,hidden_dim",
    list(itertools.product([128, 256, 512], [512, 2048, 4096])),
 )
-def test_per_tensor_quant_compare_implementations(
+def test_jit_per_tensor_quant_compare_implementations(
    num_tokens: int,
    hidden_dim: int,
 ):