tomeras91 7faba503c4 [Kernel][Mamba] Optimize Mamba2 SSD prefill Triton kernels (#35397) ...

Signed-off-by: Tomer Asida <57313761+tomeras91@users.noreply.github.com>

2026-03-04 19:47:17 +01:00

..

attention

[Core] Add All-to-All communication backend for DCP (#34883)

2026-03-04 10:01:57 -05:00

fla

[Model] Ring 2.5 (#35102)

2026-02-26 02:17:11 -08:00

fused_moe

[Hardware] Replace torch.cuda.empty_cache with torch.accelerator.empty_cache (#30681)

2026-03-04 09:49:47 +00:00

mamba

[Kernel][Mamba] Optimize Mamba2 SSD prefill Triton kernels (#35397)

2026-03-04 19:47:17 +01:00

pooler

[Bugfix] Fix corner case of sparse embedding (#33886)

2026-02-05 02:51:22 -08:00

quantization

[Hardware] Replace torch.cuda.empty_cache with torch.accelerator.empty_cache (#30681)

2026-03-04 09:49:47 +00:00

rotary_embedding

[ROCm]: Enable customop and rope+kvcache fusion for AITER RoPE (#35180)

2026-02-25 04:36:40 +00:00

__init__.py

Change the name to vLLM (#150)

2023-06-17 03:07:40 -07:00

activation.py

[XPU][1/N] Deprecate ipex and switch to vllm-xpu-kernels for xpu platform (#33379)

2026-02-02 22:46:10 -08:00

attention_layer_base.py

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

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

batch_invariant.py

[Platform] Add current_platform.num_compute_units interface (#35042)

2026-02-24 22:22:49 -08:00

conv.py

[Doc] Add developer guide for CustomOp (#30886)

2026-01-09 16:21:11 +00:00

kda.py

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

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

layernorm.py

[Bugfix] Fix dtype mismatch in RMSNormGated.forward_native() during torch.compile (#35256)

2026-03-01 15:14:46 -05:00

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

Enable bnb for multiple indices weight (#35838)

2026-03-04 01:46:47 +00:00

logits_processor.py

[Spec Decode] Reduce TP communication for speculative decoding draft token generation (#34049)

2026-02-22 14:59:16 -08:00

mla.py

[Model Bash] DeepSeek R1 BF16 Min Latency QKV A GEMM (0.5% E2E Speedup) (#34758)

2026-02-18 07:42:36 -08: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

[BugFix] Add support for MTP num_speculative_tokens > 1 with sparse MLA (#34552)

2026-03-03 07:21:57 -08:00

utils.py

[Core] Move test utility to test file (#35672)

2026-03-02 10:56:03 -05:00

vocab_parallel_embedding.py

[Doc] Add developer guide for CustomOp (#30886)

2026-01-09 16:21:11 +00:00