Gregory Shtrasberg ecff8309a3 [ROCm] Env variable to trigger custom PA (#15557) ...

Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>

2025-03-26 22:46:12 -07:00

..

mla

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

2025-03-25 09:34:59 +00: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

[V1] Enable V1 Fp8 cache for FA3 in the oracle (#15191)

2025-03-23 15:07:04 -07:00

flashinfer.py

[Misc] set single whitespace between log sentences (#13771)

2025-02-25 10:26:12 +08: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

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

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

ipex_attn.py

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

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

pallas.py

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

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

placeholder_attn.py

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

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

rocm_flash_attn.py

[ROCm] Env variable to trigger custom PA (#15557)

2025-03-26 22:46:12 -07:00

torch_sdpa.py

[CPU] Support FP8 KV cache (#14741)

2025-03-14 22:07:36 -07: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][CI] ALiBi test case in xformers multi_query_kv_attention (#11301)

2025-03-05 20:00:53 -08:00