Li, Jiang 4555143ea7 [CPU] V1 support for the CPU backend (#16441)

2025-06-03 18:43:01 -07:00

..

mla

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

abstract.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

blocksparse_attn.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

cpu_mla.py

[CPU] V1 support for the CPU backend (#16441)

2025-06-03 18:43:01 -07:00

dual_chunk_flash_attn.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

flash_attn.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

flashinfer.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

flashmla.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

hpu_attn.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

ipex_attn.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

pallas.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

placeholder_attn.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

rocm_aiter_mla.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

rocm_flash_attn.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

torch_sdpa.py

[CPU] V1 support for the CPU backend (#16441)

2025-06-03 18:43:01 -07:00

triton_mla.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

utils.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

xformers.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00