Hosang dd5fa7e04f

[ROCm][Kernel][V1] Enable AMD Radeon GPU Custom Paged Attention on v1 (#17004) ...

Signed-off-by: Hosang Yoon <hosang.yoon@amd.com>

2025-05-21 08:35:00 -07:00

..

blocksparse_attention

[Bugfix] Fix triton import with local TritonPlaceholder (#17446)

2025-05-06 17:53:09 +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][V1] Enable AMD Radeon GPU Custom Paged Attention on v1 (#17004)

2025-05-21 08:35:00 -07:00

flashmla.py

[Kernel] FlashMLA integration (#13747)

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

hpu_paged_attn.py

[Hardware][Intel-Gaudi] Support Automatic Prefix Caching on HPU (#17648)

2025-05-07 22:37:03 -07:00

ipex_attn.py

Update PyTorch to 2.7.0 (#16859)

2025-04-29 19:08:04 -07: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

[Bugfix] Fix triton import with local TritonPlaceholder (#17446)

2025-05-06 17:53:09 +08:00

rocm_aiter_mla.py

[BugFix][AMD] Compatible patch for AITER lib after 04/20 (#17912)

2025-05-13 23:05:20 -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

[Bugfix] Fix triton import with local TritonPlaceholder (#17446)

2025-05-06 17:53:09 +08:00

triton_flash_attention.py

[Bugfix] Fix triton import with local TritonPlaceholder (#17446)

2025-05-06 17:53:09 +08:00

triton_merge_attn_states.py

[Bugfix] Fix triton import with local TritonPlaceholder (#17446)

2025-05-06 17:53:09 +08:00

triton_unified_attention.py

[Kernel] update comment for KV shape in unified triton attn (#18099)

2025-05-20 11:19:34 -07:00