Pavani Majety b10f41c894

[SM100] Enable fp8 compute for prefill MLA (#30746) ...

Signed-off-by: Pavani Majety <pmajety@nvidia.com>

2025-12-22 19:15:57 +00:00

..

fla

…

fused_moe

[MoE Refactor][9/N] Use modular kernel for unquantized Triton MoE (#31052)

2025-12-22 17:34:19 +00:00

mamba

Add SpecDec support to selective_state_update (#29488)

2025-12-08 16:45:18 -05:00

quantization

[SM100] Enable fp8 compute for prefill MLA (#30746)

2025-12-22 19:15:57 +00: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

…

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

…

vocab_parallel_embedding.py

…