Kate Cheng 3d429d63a6 Enable linear deepgemm_swapAB ...

Signed-off-by: Kate Cheng <yunhsuanc@nvidia.com>

2025-12-24 11:19:39 -08:00

..

fla

…

fused_moe

[MoE Refactor][7/N] AITER MK (#31102)

2025-12-22 16:42:58 -07:00

mamba

[Mamba] - Consolidate Mambas Attention Logic (#28133)

2025-12-23 21:57:00 +01:00

quantization

Enable linear deepgemm_swapAB

2025-12-24 11:19:39 -08:00

rotary_embedding

[CustomOp][Refactor] Extract common methods for ApplyRotaryEmb CustomOp (#31021)

2025-12-19 22:16:09 +08:00

__init__.py

…

activation.py

[ROCm] Fallback pytorch GELU with tanh approximation to GELU() (#29244)

2025-12-02 02:02:22 +00:00

attention_layer_base.py

[Attention] Update attention imports (#29540)

2025-11-27 11:19:09 -05:00

batch_invariant.py

[Bug] Fix batch invariant in torch 2.10 (#30907)

2025-12-18 07:27:51 -08:00

conv.py

2.9.1 PyTorch release update (#28495)

2025-12-17 12:20:22 -08:00

kda.py

[Model][Mamba] Add selector for mamba attention backend and make it pluggable for other device (#26487)

2025-11-19 16:24:55 +00:00

layernorm.py

…

lightning_attn.py

[CI/Build][Kernel][AMD] Move extra dim to after load in _fwd_kv_parallel in lighting_attn.py (#29132)

2025-11-21 11:53:09 -05:00

linear.py

[Feature]: Support NVIDIA ModelOpt HF FP8 variants FP8_PER_CHANNEL_PER_TOKEN and FP8_PB_WO in vLLM (#30957)

2025-12-21 22:34:49 -05:00

logits_processor.py

…

mla.py

fix#30092 Kimi-Linear model loading failure with missing indexer_rotary_emb (#30093)

2025-12-05 20:55:09 -08:00

pooler.py

[Model][6/N] Improve all pooling task | Support chunked prefill with ALL pooling (#27145)

2025-12-04 13:44:15 +00:00

resampler.py

…

utils.py

[platform] Move get_cu_count to utils (#27005)

2025-11-13 08:48:47 +08:00

vocab_parallel_embedding.py

…