fix sgl-kernel unit tests (#5666)

2025-04-23 01:18:30 -07:00
parent e62c49557d
commit 15fabcc07f
9 changed files with 313 additions and 0 deletions
--- a/sgl-kernel/python/sgl_kernel/init.py
+++ b/sgl-kernel/python/sgl_kernel/init.py
@@ -41,6 +41,7 @@ from sgl_kernel.gemm import (
    sgl_per_token_group_quant_int8,
    sgl_per_token_quant_fp8,
 )
+from sgl_kernel.grammar import apply_token_bitmask_inplace_cuda
 from sgl_kernel.moe import (
    fp8_blockwise_scaled_grouped_mm,
    moe_align_block_size,