Johnsonms 7cf0551014 Migrate norm kernels to FlashInfer JIT implementation (#18871)

2026-03-10 14:56:07 +08:00

..

quantization

…

testing

…

__init__.py

…

_fa4_interface.py

…

allreduce.py

…

attention.py

…

cutlass_moe.py

…

elementwise.py

Migrate norm kernels to FlashInfer JIT implementation (#18871)

2026-03-10 14:56:07 +08:00

expert_specialization.py

…

flash_attn.py

…

flash_mla.py

…

gemm.py

…

grammar.py

…

kvcacheio.py

…

load_utils.py

…

mamba.py

…

memory.py

…

moe.py

…

sampling.py

…

scalar_type.py

…

sparse_flash_attn.py

…

spatial.py

…

speculative.py

…

test_utils.py

…

top_k.py

…

utils.py

…

version.py

…