xuebwang 81503550f3 [ROCm] Disable AITER FP8/FP4 BMM for MLA by default (#53123) ...

Signed-off-by: xuebwang-amd <xuebwang@amd.com>

2026-10-14 10:46:02 +00:00

..

__init__.py

[Bugfix][Model Loader] Defer post-load attention weight processing (#49519)

2026-08-10 19:16:57 -07:00

attention.py

[Kernel] Resubmit PR 48666 - Gemma4 FP8 KV FA4 head dim 512 backend selection (#53175)

2026-09-23 20:29:06 -07:00

chunked_local_attention.py

[Attention]: Use KVCacheSpec for AttentionMetadataBuilder type hints (#50148)

2026-07-31 11:20:18 +00:00

cross_attention.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

encoder_only_attention.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

kv_transfer_utils.py

[compile] Allow strings in custom ops without regressing compilation times (#38123)

2026-04-10 07:26:37 +00:00

mla_attention.py

[ROCm] Disable AITER FP8/FP4 BMM for MLA by default (#53123)

2026-10-14 10:46:02 +00:00

mm_encoder_attention.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

prefill_prefix_lm_attention.py

[Model] Add HrmTextForCausalLM (Hierarchical Reasoning Model — Text) (#43098)

2026-06-16 22:41:41 +08:00

rswa_attention.py

[Model] Support Unlimited OCR (#46564)

2026-06-27 23:09:18 -07:00

sparse_mla_attention.py

[HiSparse] Read GPU-resident KV directly for mixed and BF16 prefills (#60417)

2026-10-13 16:40:20 -04:00

sparse_mla_mask.py

[Attention] Vectorize sparse MLA mask loads (#52217)

2026-08-18 20:32:41 -07:00

static_sink_attention.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00