Andy Lo 322d2a27d6 [BugFix] Minor: logger import in attention backend (#13706) ...

Signed-off-by: Andy Lo <andy@mistral.ai>

2025-02-22 16:51:13 -08:00

..

mla

[Attention] MLA with chunked prefill (#12639)

2025-02-21 15:30:12 -08:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

abstract.py

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

2025-02-02 11:58:18 -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

[bugfix] fix early import of flash attention (#12959)

2025-02-09 00:06:56 +08:00

flashinfer.py

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

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

hpu_attn.py

[Hardware][Gaudi][Feature] Support Contiguous Cache Fetch (#12139)

2025-02-18 19:40:19 -08:00

ipex_attn.py

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

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

openvino.py

[Bugfix] Fix OpenVINO model runner (#12750)

2025-02-04 22:42:46 -08:00

pallas.py

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

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

placeholder_attn.py

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

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

rocm_flash_attn.py

[V1][Kernel] Refactor the prefix_prefill kernel so that the caller no longer has to pass in the context lengths (#13095)

2025-02-22 05:25:41 -08:00

torch_sdpa.py

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

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

triton_mla.py

[Attention] MLA with chunked prefill (#12639)

2025-02-21 15:30:12 -08:00

utils.py

[BugFix] Minor: logger import in attention backend (#13706)

2025-02-22 16:51:13 -08:00

xformers.py

[V1][Kernel] Refactor the prefix_prefill kernel so that the caller no longer has to pass in the context lengths (#13095)

2025-02-22 05:25:41 -08:00