Luka Govedič e1744502c2 [FP8] Refactor apply_fp8_linear and apply_fp8_linear_generic into an object (#14390) ...

Signed-off-by: luka <luka@neuralmagic.com>

2025-03-07 05:20:16 +00:00

..

mla

[FP8] Refactor apply_fp8_linear and apply_fp8_linear_generic into an object (#14390)

2025-03-07 05:20:16 +00:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

abstract.py

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

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

blocksparse_attn.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

flash_attn.py

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

2025-03-06 14:18:29 -08: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

openvino.py

[Bugfix] Fix OpenVINO model runner (#12750)

2025-02-04 22:42:46 -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] Faster Custom Paged Attention kernels (#12348)

2025-03-03 09:24:45 -08:00

torch_sdpa.py

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

2025-03-06 14:18:29 -08: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

[Hardware] Update the flash attn tag to support Blackwell (#14244)

2025-03-05 22:01:37 -08:00

xformers.py

[Bugfix][CI] ALiBi test case in xformers multi_query_kv_attention (#11301)

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