Aleksandr MalyshevandAleksandr Malyshev 64862d106e [ROCM][AMD][TRITON] Halving warps number for fw_prefill to reduce spilling (#12713) ...

Signed-off-by: Aleksandr Malyshev <maleksan@amd.com>
Co-authored-by: Aleksandr Malyshev <maleksan@amd.com>

2025-02-05 03:58:22 +00:00

..

blocksparse_attention

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

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

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

hpu_paged_attn.py

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

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

ipex_attn.py

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

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

nki_flash_attn.py

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

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

paged_attn.py

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

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

prefix_prefill.py

[ROCM][AMD][TRITON] Halving warps number for fw_prefill to reduce spilling (#12713)

2025-02-05 03:58:22 +00:00

triton_decode_attention.py

[Perf] Mem align KV caches for CUDA devices (MLA perf improvement) (#12676)

2025-02-04 18:22:24 -08:00

triton_flash_attention.py

[Misc] Fix improper placement of SPDX header in scripts (#12694)

2025-02-03 11:16:59 -08:00