Andrii SkliarandAndrii Skliar adebc41b7e [Mamba] Add FlashInfer ReplaySSM backend (#52506) ...

Signed-off-by: Andrii Skliar <askliar@nvidia.com>
Signed-off-by: Andrii Skliar <andreyws96@gmail.com>
Co-authored-by: Andrii Skliar <askliar@nvidia.com>

2026-09-01 10:26:43 -04:00

..

attention

[Attention][DSA] Enable W4A16 DSA (#51724)

2026-09-01 12:37:47 +08:00

fused_moe

[CPU][Zen] Route Int8 MoE inference through zentorch on AMD (#44834)

2026-09-01 09:31:31 +00:00

fusion

Fix docs build on main (#45536)

2026-06-13 08:53:23 -07:00

hpc

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

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

mamba

[Mamba] Add FlashInfer ReplaySSM backend (#52506)

2026-09-01 10:26:43 -04: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

[Bugifx][INC] Fix INC quantization method selection for non-quantized layers (#47237)

2026-09-01 10:40:47 +00:00

rotary_embedding

[XPU] follow cuda path for mrope on XPU (#53201)

2026-08-21 10:13:27 +00: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

[Refactor] Refactor batch invariance folder (#53619)

2026-08-24 14:31:36 -07: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

[XPU] Ensure unquantized linear weight is N-contiguous (#53536)

2026-08-31 16:01:24 +08:00

logits_processor.py

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

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

mhc.py

[ROCm][Perf] Fuse DeepSeek-V4 mHC post/pre and RMSNorm with AITER (#52737)

2026-08-20 04:51:04 +00:00

mla.py

[New model] Kimi K3 (#50000)

2026-07-30 03:49:58 -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.py

[Perf][Kernel] Tune cooperative topk for medium batch-sizes (#53382)

2026-08-31 13:44:38 -07:00

utils.py

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

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

vocab_parallel_embedding.py

[kernel] Fused embedding kernel (#53677)

2026-08-31 11:34:14 +00:00