wangxiyuan 5536fc0c01 [Misc] Replace mamba_type string literals with MambaAttentionBackendEnum (#41188) ...

Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>

2026-05-11 03:59:36 +00:00

..

attention

[Attention] Minor refactor: layer takes ownership of the MLA prefill backend (#41744)

2026-05-05 23:22:41 +00:00

fla

[Bugfix] Fix GDN KKT precision loss on Hopper GPUs by aligning tl.dot operand layout with WGMMA (#42076)

2026-05-09 13:08:46 +00:00

fused_moe

[MoE] Move various experts classes to fused_moe/experts/ (#41979)

2026-05-11 07:54:33 +08:00

mamba

[Misc] Replace mamba_type string literals with MambaAttentionBackendEnum (#41188)

2026-05-11 03:59:36 +00:00

pooler

[Perf][2/n] Eliminate GPU<->CPU syncs in pooling code (#41433)

2026-05-05 02:43:25 +00:00

quantization

[MoE] Move various experts classes to fused_moe/experts/ (#41979)

2026-05-11 07:54:33 +08:00

rotary_embedding

[Bugfix] Fix repeated DSv4 RoPE cache initialization (#41148)

2026-04-29 20:29:55 +08:00

__init__.py

…

activation.py

[Models][Gemma3/Gemma4] Support hidden_act variants in gated MLP (#40588)

2026-05-08 11:29:11 -07:00

attention_layer_base.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

batch_invariant.py

[BE][Torch 2.12] Remove workaround code for fixed cublas issue (#40845)

2026-04-28 21:07:24 -07:00

conv.py

[Misc] Always use forward_mulmat for Conv3d on newer versions of torch. (#38487)

2026-03-30 05:39:41 +00:00

deepseek_compressor.py

[New Model][ROCm] Add AMD support for DeepSeek V4 (#40871)

2026-05-05 08:55:37 -07:00

deepseek_v4_attention.py

[New Model][ROCm] Add AMD support for DeepSeek V4 (#40871)

2026-05-05 08:55:37 -07:00

kda.py

[Misc] Replace mamba_type string literals with MambaAttentionBackendEnum (#41188)

2026-05-11 03:59:36 +00:00

layernorm.py

[vLLM IR] 2/N fused_add_rms_norm and maybe_inplace overload (#36823)

2026-05-01 23:41:15 -04:00

lightning_attn.py

[Bugfix] Use null block (0) for padded block table entries (#35431)

2026-03-30 14:02:51 -07:00

linear.py

[Quantization] Add ModelOpt NVFP4 W4A16 (4-bit weights, fp16/bf16 activations) support (#41769)

2026-05-09 21:15:50 +00:00

logits_processor.py

[PluggableLayer][3/N] Apply PluggableLayer to llm_head and vocab embedding layer (#33465)

2026-04-10 16:12:59 +08:00

mhc.py

add fused mhc_post_pre kernel (#41536)

2026-05-10 19:56:52 -07:00

mla.py

[Feature]: IndexCache support for DSA models (#37735)

2026-04-29 15:15:35 -04: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

[New Model][ROCm] Add AMD support for DeepSeek V4 (#40871)

2026-05-05 08:55:37 -07:00

utils.py

[DeepSeek] Use torch.mm for bf16xbf16->fp32 gemm (#41300)

2026-04-30 16:28:57 -07:00

vocab_parallel_embedding.py

Bug/test eagle dp v0 (#38938)

2026-04-13 20:50:08 +00:00