rasmith 8e4b351a0c

[Kernel][Triton][FP8] Adding fp8 and variable length sequence support to Triton FAv2 kernel (#12591) ...

Signed-off-by: Randall Smith <Randall.Smith@amd.com>

2025-04-27 00:35:08 +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

chunked_prefill_paged_decode.py

[ROCM][KERNEL] Paged attention for V1 (#15720)

2025-04-02 19:48:00 -07:00

flashmla.py

[Kernel] FlashMLA integration (#13747)

2025-02-27 10:35:08 +08:00

hpu_paged_attn.py

[Hardware][Intel-Gaudi] Update hpu-extension and update bucketing system for HPU device (#17186)

2025-04-26 05:55:14 -07:00

ipex_attn.py

[CPU] Upgrade CPU backend to torch-2.6 (#13381)

2025-03-12 10:41:13 +00:00

merge_attn_states.py

[Kernel] support merge_attn_states CUDA kernel, 3x speedup (#16173)

2025-04-11 06:50:50 -06:00

nki_flash_attn.py

[BugFix] fix some typos found by typos. (#16314)

2025-04-09 03:43:59 -07:00

paged_attn.py

[ROCM][KERNEL] Paged attention for V1 (#15720)

2025-04-02 19:48:00 -07:00

prefix_prefill.py

[Kernel][ROCM] Upstream prefix prefill speed up for vLLM V1 (#13305)

2025-04-22 19:11:56 -07:00

rocm_aiter_mla.py

[FEAT][ROCm]: Support AITER MLA (#15893)

2025-04-22 09:31:13 -07:00

rocm_aiter_paged_attn.py

[FEAT][ROCm] Integrate Paged Attention Kernel from AITER (#15001)

2025-04-22 02:46:28 -07:00

triton_decode_attention.py

[Misc] remove warning if triton>=3.2.0 (#16553)

2025-04-14 02:39:47 -07:00

triton_flash_attention.py

[Kernel][Triton][FP8] Adding fp8 and variable length sequence support to Triton FAv2 kernel (#12591)

2025-04-27 00:35:08 +00:00

triton_merge_attn_states.py

[Kernel] Remove redundant Exp calculations (#16123)

2025-04-15 12:58:37 +00:00