Wei-Yu Lin 6125a172f9 Remove pallas.py as this is migrate to tpu-inference ...

Signed-off-by: Wei-Yu Lin <weiyulin@google.com>

2025-12-25 00:51:58 +00:00

..

mla

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

2025-12-22 18:15:33 -08:00

__init__.py

…

cpu_attn.py

[CPU] Support for Whisper (#30062)

2025-12-10 04:58:42 -08:00

flash_attn.py

[Attention] Cache attention metadata builds across hybrid KV-cache groups (#29627)

2025-12-16 17:10:16 -05:00

flashinfer.py

[Cleanup] Refactor FlashInferMetadataBuilder (#29128)

2025-12-18 14:45:30 -08:00

flex_attention.py

[Fix][FlexAttention] return max logical block index to handle reused blocks (#30915)

2025-12-18 06:42:21 +00:00

gdn_attn.py

[Perf] Remove blocking copy in GDN Attention (#31167)

2025-12-22 14:25:22 -08:00

linear_attn.py

[Attention] add _cudagraph_support for linear attention (#28934)

2025-11-25 12:25:20 +08:00

mamba1_attn.py

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

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

mamba2_attn.py

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

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

mamba_attn.py

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

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

rocm_aiter_fa.py

[ROCm][Attention] Sliding window support for AiterFlashAttentionBackend (#29234)

2025-11-30 11:31:50 +00:00

rocm_aiter_unified_attn.py

[ROCm] Support for Whisper v1 with Aiter Unified Attention and Aiter Flash Attention (#28376)

2025-11-24 03:26:00 +00:00

rocm_attn.py

[ROCm][CI] Reduce Flakiness For test_async_scheduling Using ROCM_ATTN With FP32 (#30811)

2025-12-18 10:27:37 +08:00

short_conv_attn.py

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

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

tree_attn.py

[Attention] Refactor FA block_size limitations to hybrid models only (#29084)

2025-11-22 06:38:44 -08:00

triton_attn.py

[v1] Add PrefixLM support to TritonAttention backend (#30386)

2025-12-17 16:05:24 -08:00

utils.py

adapt voxtral (#31095)

2025-12-23 05:31:55 -08:00