Lucas Wilkinson d8bccde686 [BugFix] Fix vllm_flash_attn install issues (#17267) ...

Signed-off-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>
Co-authored-by: Jee Jee Li <pandaleefree@gmail.com>
Co-authored-by: Aaron Pham <contact@aarnphm.xyz>

2025-04-27 17:27:56 -07:00

..

mla

[BugFix] Fix vllm_flash_attn install issues (#17267)

2025-04-27 17:27:56 -07:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

abstract.py

Allocate kv_cache with stride order (#16605)

2025-04-25 22:03:31 -07: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 vllm_flash_attn install issues (#17267)

2025-04-27 17:27:56 -07:00

flashinfer.py

[Bugfix] Get a specific type of layer from forward context (#17222)

2025-04-27 00:58:05 -07: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

[Hardware][Intel-Gaudi] Update hpu-extension and update bucketing system for HPU device (#17186)

2025-04-26 05:55:14 -07:00

ipex_attn.py

[XPU][Bugfix] minor fix for XPU (#15591)

2025-04-22 00:02:57 +08: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_aiter_mla.py

[FEAT][ROCm]: Support AITER MLA (#15893)

2025-04-22 09:31:13 -07:00

rocm_flash_attn.py

[Quantization][FP8] Add support for FP8 models with input_scale for output projection and QK quantization (#15734)

2025-04-25 00:45:02 -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] Update to lastest FA3 code (#13111)

2025-04-17 15:14:07 -07:00

xformers.py

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

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