Lucas Wilkinson 2d9ee28cab

[CI/Test Fix] Fix CP tests on Blackwell (#29338) ...

Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>

2025-11-24 20:55:57 -08:00

..

__init__.py

…

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

[CI/Test Fix] Fix CP tests on Blackwell (#29338)

2025-11-24 20:55:57 -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

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

[Core] Deprecate xformers (#29262)

2025-11-24 04:18:55 +00:00