Wei Zhao a69402aaa8 [Perf] Integrate FlashInfer KDA kernels (#55364) ...

Signed-off-by: wzhao18 <wzhao18.sz@gmail.com>

2026-09-07 20:29:30 -07:00

..

attention

[Kernel] Remove unused fake implementation (#55535)

2026-09-06 17:54:32 -07:00

fused_moe

[Kernel][Perf] Tune H20 block-FP8 MoE low-batch configs (+21%) (#54668)

2026-09-08 03:20:21 +00:00

fusion

[Fusion] Manual ActivationQuantFusionPass initial application (#51415)

2026-09-03 13:37:31 -04:00

hpc

[Kernel] Remove unused fake implementation (#55535)

2026-09-06 17:54:32 -07:00

mamba

[Perf] Integrate FlashInfer KDA kernels (#55364)

2026-09-07 20:29:30 -07:00

minimax_rms_norm

[ROCm][MiniMax-M2] Dispatch fused QK-norm + AllReduce via AITER (#44849)

2026-07-14 03:10:16 +00:00

pooler

[Perf] Avoid h2d copies from non-pinned CPU tensors (#54299)

2026-08-29 12:06:30 +08:00

quantization

[Bugfix][Quantization][MoE] Normalise an unset group_size on the compressed-tensors WNA16 MoE path (#53163)

2026-09-08 09:42:46 +08:00

rotary_embedding

[Kernel] Remove unused fake implementation (#55535)

2026-09-06 17:54:32 -07:00

__init__.py

…

activation.py

[XPU] Route activation CustomOps to SYCL kernels (#53734)

2026-09-01 09:19:52 +08:00

attention_layer_base.py

[3/N][KV-Cache Layout Refactor] Standardize Mamba cache; drop get_transfer_cache_regions (#44456)

2026-07-21 09:16:15 +00:00

conv.py

[Refactor] Remove multiple dead codes (#50285)

2026-08-03 10:59:08 -04:00

fused_allreduce_gemma_rms_norm.py

buffer size insuffient Dspark sd for FlashInfer MNNVL allreduce (#50932)

2026-08-27 11:01:08 -04:00

fused_embed_norm.py

Replicated embedding and norm fusion for DSV3 flat model (#48484)

2026-08-18 14:53:21 +08:00

fused_qk_norm_rope.py

[Kernel][Perf] Enable fused QK-norm + partial MRoPE + gate for Qwen3.6 (#52676)

2026-08-24 18:28:49 -07:00

layernorm.py

[XPU] Add fused GemmaRMSNorm path for eager execution (#53678)

2026-09-03 06:11:20 +08:00

lightning_attn.py

[Kernel][Test] Extend lightning_attn and awq_triton kernel tests to XPU (#43307)

2026-06-04 14:25:59 -04:00

linear.py

[Quantization] Support online quantization with partially pre-quantized checkpoints (#51392)

2026-09-04 11:46:06 -07:00

logits_processor.py

[Hy4] support Hy4-preview model (#54160)

2026-08-29 12:07:28 +08:00

mhc.py

[Model] add GLM-5.3-Flash support (#53906)

2026-09-03 09:40:35 -07:00

mla.py

[Model] add GLM-5.3-Flash support (#53906)

2026-09-03 09:40:35 -07:00

resampler.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

sparse_attn_indexer_kpool.py

[Model] add GLM-5.3-Flash support (#53906)

2026-09-03 09:40:35 -07:00

sparse_attn_indexer.py

[2/N][warmup][DSv4] Migrate sequence and DCP kernels (#53564)

2026-09-07 09:35:54 -04:00

utils.py

[Bugfix][CPU] Fix several bugs (#54042)

2026-08-31 15:33:06 +08:00

vocab_parallel_embedding.py

[Quantization] Support online quantization with partially pre-quantized checkpoints (#51392)

2026-09-04 11:46:06 -07:00