Augusto Yao 9726e64530 bugfix: correct attn output with base 2 or e (#28840) ...

Signed-off-by: augusto.yjh <augusto.yjh@antgroup.com>

2025-11-29 07:52:12 +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

bugfix: correct attn output with base 2 or e (#28840)

2025-11-29 07:52:12 +08: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

[V0 deprecation] Clean up legacy paged attention helper functions (#28043)

2025-11-28 16:44:33 +00: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

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

[Performance][MLA][ROCm] Remove redundant D2D copy in deepseek (#27457)

2025-11-26 12:45:28 +08: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

Remove upstream fa checks (#29471)

2025-11-28 05:52:42 -08:00