frida-andersson 71fc70d3ae [ROCm][DSv4] Fix sparse-indexer logits collapse on gfx950/gfx942 (#50455) ...

Signed-off-by: Frida Andersson <fanderss@amd.com>
Signed-off-by: frida-andersson <fanderss@amd.com>

2026-09-18 16:57:44 +00:00

..

flydsl_kernels

[Kernel][ROCm][Perf] FlyDSL decode-attention kernel for 4-bit TurboQuant KV cache (#47896)

2026-08-11 08:45:39 -07:00

turboquant_soa

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

__init__.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

chunked_prefill_paged_decode.py

[ROCm] [bugfix] Chunked prefill paged decode masked load perf (#50017)

2026-08-12 23:04:01 +00:00

common.py

Add pydocstyle to the ruff rules (#52136)

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

cp_common.py

[refactor] consolidate cp attn ops (#52839)

2026-08-19 18:04:11 -07:00

dcp.py

Add pydocstyle to the ruff rules (#52136)

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

flashmla.py

Add pydocstyle to the ruff rules (#52136)

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

flydsl_turboquant_decode.py

[Kernel][ROCm][Perf] FlyDSL decode-attention kernel for 4-bit TurboQuant KV cache (#47896)

2026-08-11 08:45:39 -07:00

int4_per_token_head.py

[Feature] Triton INT4 per-token-head KV cache quantization (#40835)

2026-06-24 10:21:25 +00:00

merge_attn_states.py

Add pydocstyle to the ruff rules (#52136)

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

metadata.py

[Perf] Fuse DSV4.1 input metadata preparation with Triton (#56562)

2026-09-12 07:20:04 -07:00

paged_attn.py

[XPU][9/N] clean up existing ipex code/doc (#34111)

2026-02-11 00:27:15 -08:00

pcp.py

[refactor] consolidate cp attn ops (#52839)

2026-08-19 18:04:11 -07:00

prefix_prefill.py

[Kernel] Make prefix-prefill tiling independent of the KV page size (#54194)

2026-09-01 09:19:55 +08:00

rocm_aiter_mla_merge.py

[ROCm][MLA][DCP] Support causal multi-token verification (#51705)

2026-08-31 10:42:28 -07:00

rocm_aiter_mla_sparse.py

[ROCm][DSv4] Fix sparse-indexer logits collapse on gfx950/gfx942 (#50455)

2026-09-18 16:57:44 +00:00

triton_attention_helpers.py

[Kernel][Gemma4] Prune Triton sliding-window tiles for multimodal prefixes (#53147)

2026-08-31 22:21:22 +00:00

triton_decode_attention.py

Add pydocstyle to the ruff rules (#52136)

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

triton_merge_attn_states.py

[Attention][MLA] Per-request scheduling for MLA chunked context (#50613)

2026-08-06 16:45:44 +00:00

triton_prefill_attention.py

Add pydocstyle to the ruff rules (#52136)

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

triton_reshape_and_cache_flash.py

[2/N][KV-Cache Layout Refactor] Pack K/V into the content dim across attention backends (#44455)

2026-07-11 11:11:16 -04:00

triton_turboquant_decode.py

[Attention][TurboQuant] Share dequant buffers, eliminate float16_copy (#40941)

2026-04-27 13:48:36 +08:00

triton_turboquant_store.py

[2/N][KV-Cache Layout Refactor] Pack K/V into the content dim across attention backends (#44455)

2026-07-11 11:11:16 -04:00

triton_unified_attention_diffkv.py

[Model] Add DiffusionGemma Support (#45163)

2026-06-11 22:17:35 -07:00

triton_unified_attention.py

[Kernel][Gemma4] Prune Triton sliding-window tiles for multimodal prefixes (#53147)

2026-08-31 22:21:22 +00:00

vit_attn_wrappers.py

Add pydocstyle to the ruff rules (#52136)

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

xpu_mla_sparse.py

Add pydocstyle to the ruff rules (#52136)

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