LICENSE
README.md
pyproject.toml
setup_hip.py
csrc/common_extension_rocm.cc
csrc/allreduce/custom_all_reduce.hip
csrc/allreduce/quick_all_reduce.hip
csrc/elementwise/activation.hip
csrc/grammar/apply_token_bitmask_inplace_cuda.hip
csrc/kvcacheio/transfer.hip
csrc/moe/moe_align_kernel.hip
csrc/moe/moe_topk_softmax_kernels.hip
csrc/speculative/eagle_utils.hip
python/sgl_kernel/__init__.py
python/sgl_kernel/allreduce.py
python/sgl_kernel/attention.py
python/sgl_kernel/cutlass_moe.py
python/sgl_kernel/elementwise.py
python/sgl_kernel/flash_attn.py
python/sgl_kernel/fused_moe.py
python/sgl_kernel/gemm.py
python/sgl_kernel/grammar.py
python/sgl_kernel/kvcacheio.py
python/sgl_kernel/marlin.py
python/sgl_kernel/memory.py
python/sgl_kernel/moe.py
python/sgl_kernel/sampling.py
python/sgl_kernel/scalar_type.py
python/sgl_kernel/sparse_flash_attn.py
python/sgl_kernel/spatial.py
python/sgl_kernel/speculative.py
python/sgl_kernel/top_k.py
python/sgl_kernel/utils.py
python/sgl_kernel/version.py
python/sgl_kernel.egg-info/PKG-INFO
python/sgl_kernel.egg-info/SOURCES.txt
python/sgl_kernel.egg-info/dependency_links.txt
python/sgl_kernel.egg-info/top_level.txt
python/sgl_kernel/testing/__init__.py
python/sgl_kernel/testing/rotary_embedding.py
tests/test_activation.py
tests/test_apply_token_bitmask_inplace.py
tests/test_awq_dequant.py
tests/test_bmm_fp8.py
tests/test_custom_allreduce.py
tests/test_cutlass_mla.py
tests/test_cutlass_w4a8_moe_mm.py
tests/test_dsv3_fused_a_gemm.py
tests/test_dsv3_router_gemm.py
tests/test_ep_moe_post_reorder_kernel.py
tests/test_ep_moe_pre_reorder_kernel.py
tests/test_ep_moe_silu_and_mul_kernel.py
tests/test_flash_attention.py
tests/test_fp4_gemm.py
tests/test_fp4_quantize.py
tests/test_fp8_blockwise_gemm.py
tests/test_fp8_blockwise_moe.py
tests/test_fp8_gemm.py
tests/test_gptq_kernel.py
tests/test_int8_gemm.py
tests/test_kvcacheio.py
tests/test_lightning_attention_decode.py
tests/test_marlin_gemm.py
tests/test_marlin_repack.py
tests/test_merge_state.py
tests/test_merge_state_v2.py
tests/test_moe_align.py
tests/test_moe_fused_gate.py
tests/test_moe_topk_softmax.py
tests/test_mscclpp.py
tests/test_norm.py
tests/test_per_tensor_quant_fp8.py
tests/test_per_token_group_quant_8bit.py
tests/test_per_token_quant_fp8.py
tests/test_qserve_w4a8_per_chn_gemm.py
tests/test_qserve_w4a8_per_group_gemm.py
tests/test_rotary_embedding.py
tests/test_sampling.py
tests/test_sparse_flash_attn.py