DefTruth e9528f6dc6

[Kernel] support merge_attn_states CUDA kernel, 3x speedup (#16173) ...

Signed-off-by: DefTruth <qiustudent_r@163.com>

2025-04-11 06:50:50 -06:00

..

mla

[Kernel] support merge_attn_states CUDA kernel, 3x speedup (#16173)

2025-04-11 06:50:50 -06:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

2024-03-25 04:39:33 +00:00

abstract.py

[Attention] Flash Attention 3 - fp8 (#14570)

2025-03-20 01:14:20 -04:00

blocksparse_attn.py

fix some typos : supported_head_sizes (#14627)

2025-03-11 10:38:24 -07:00

cpu_mla.py

[Kernel][CPU] CPU MLA (#14744)

2025-03-25 09:34:59 +00:00

flash_attn.py

[BugFix] fix some typos found by typos. (#16314)

2025-04-09 03:43:59 -07:00

flashinfer.py

Add warning for Attention backends that do not support irope yet (#16212)

2025-04-08 03:59:26 +00:00

flashmla.py

[V1][Bugfix] Standardize quantized kv cache rejection for attention backends (#14221)

2025-03-06 14:18:29 -08:00

hpu_attn.py

[BugFix] fix some typos found by typos. (#16314)

2025-04-09 03:43:59 -07:00

ipex_attn.py

Add warning for Attention backends that do not support irope yet (#16212)

2025-04-08 03:59:26 +00:00

pallas.py

Add warning for Attention backends that do not support irope yet (#16212)

2025-04-08 03:59:26 +00:00

placeholder_attn.py

[Model] IBM/NASA Prithvi Geospatial model (#12830)

2025-02-11 20:34:30 -08:00

rocm_flash_attn.py

[Bug] [ROCm] Fix Llama 4 Enablement Bug on ROCm: V0 ROCmFlashAttentionImpl and Triton Fused MoE bugs (#16198)

2025-04-08 19:12:34 -07:00

torch_sdpa.py

Add warning for Attention backends that do not support irope yet (#16212)

2025-04-08 03:59:26 +00:00

triton_mla.py

[V1][Bugfix] Standardize quantized kv cache rejection for attention backends (#14221)

2025-03-06 14:18:29 -08:00

utils.py

[Attention] Flash Attention 3 - fp8 (#14570)

2025-03-20 01:14:20 -04:00

xformers.py

[BugFix] fix some typos found by typos. (#16314)

2025-04-09 03:43:59 -07:00