Bram Wasti b2f78cbad4

[small][batch invariance] Rename the env and internal flags to simplify usage (#26855) ...

Signed-off-by: Bram Wasti <bwasti@meta.com>

2025-10-16 21:40:25 +00:00

..

mla

[small][batch invariance] Rename the env and internal flags to simplify usage (#26855)

2025-10-16 21:40:25 +00:00

__init__.py

[V1] Implement vLLM V1 [1/N] (#9289)

2024-10-22 01:24:07 -07:00

cpu_attn.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

flash_attn.py

[small][batch invariance] Rename the env and internal flags to simplify usage (#26855)

2025-10-16 21:40:25 +00:00

flashinfer.py

[small][batch invariance] Rename the env and internal flags to simplify usage (#26855)

2025-10-16 21:40:25 +00:00

flex_attention.py

[small][batch invariance] Rename the env and internal flags to simplify usage (#26855)

2025-10-16 21:40:25 +00:00

gdn_attn.py

[PERF] Qwen3-next MTP speedup (change bool mask indexing to index_select / index_copy to reduce d2h) (#26437)

2025-10-16 12:18:31 +08:00

linear_attn.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

mamba1_attn.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

mamba2_attn.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

mamba_attn.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

pallas.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

rocm_aiter_fa.py

remove attn output view kernel (#26680)

2025-10-14 22:53:10 +00:00

rocm_aiter_unified_attn.py

remove attn output view kernel (#26680)

2025-10-14 22:53:10 +00:00

rocm_attn.py

remove attn output view kernel (#26680)

2025-10-14 22:53:10 +00:00

short_conv_attn.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

tree_attn.py

remove attn output view kernel (#26680)

2025-10-14 22:53:10 +00:00

triton_attn.py

Move query quantization to attention layer for Flashinfer & Triton. (#26534)

2025-10-15 19:01:38 -04:00

utils.py

[DCP] Support Decode Context Parallel (DCP) for GQA with FlashAttention (#24864)

2025-10-14 13:07:50 +00:00

xformers.py

remove attn output view kernel (#26680)

2025-10-14 22:53:10 +00:00