Pleaplusone 06c20c9904 [ROCm] Add AMD GPU support on Deepseek v3.2 and SparseMLA (#26670) ...

Signed-off-by: ganyi <ygan@amd.com>

2025-11-20 02:54:01 -08:00

..

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

chunked_prefill_paged_decode.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

common.py

[Feature] Prefill Context Parallel (PCP) basic support (#28718)

2025-11-19 15:52:44 -05:00

flashmla.py

[Bugfix] fixes the decoding metadata of dense mla's fp8 kvcache. (#27144)

2025-10-21 18:27:03 +00:00

merge_attn_states.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

paged_attn.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

pallas_kv_cache_update.py

[Chore]:Extract math and argparse utilities to separate modules (#27188)

2025-10-26 04:03:32 -07:00

prefix_prefill.py

[Misc] Remove unused attention prefix prefill ops functions (#26971)

2025-11-11 18:26:04 +00:00

rocm_aiter_mla_sparse.py

[ROCm] Add AMD GPU support on Deepseek v3.2 and SparseMLA (#26670)

2025-11-20 02:54:01 -08:00

rocm_aiter_paged_attn.py

[Chore]:Extract math and argparse utilities to separate modules (#27188)

2025-10-26 04:03:32 -07:00

triton_decode_attention.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

triton_merge_attn_states.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

triton_reshape_and_cache_flash.py

[Bugfix][CI/Test][Spec Decode] Fix illegal memory access in offline_inference/spec_decode.py (Issue 27619) (#28432)

2025-11-13 22:34:01 -08:00

triton_unified_attention.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

vit_attn_wrappers.py

[Bugfix][Qwen][Multimodal] Move Qwen2_5_vl sdpa to custom op and reenable compile (#27764)

2025-11-03 11:12:15 -08:00