jiangyon.ren f6d2ca82b7 [CK_TILE][FMHA] Add sparse attention VSA (#3341) ...

* add sparse attention VSA

* fix the pre-commit

* Add jenga test and pre-commit

* add bf16 for vsa

* add jenga support bf16

* remove lse arg

* split kernel code to block & kernel

* fix the pre-commit

* fix the pre-commit

* fix the copyrights

* fix the copyright

* fix the copyright & rename block to pipeline

* fix the copyright and pipeline

* remove lse & dropout & add fmt

* fix the jenga&VSA code review

* remove the useless code & resolved the comments

* remove useless code

* remove useless code

* Clean up code

* Remove more unused code

* Re-format .hpp

* Refactor codegen scripts

---------

Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>
Co-authored-by: asleepzzz <hanwen.chang@amd.com>

[ROCm/composable_kernel commit: 4d2f8c111e]

2026-01-31 00:59:47 +08:00

..

add_rmsnorm2d_rdquant

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

batched_contraction

[CK Tile] batched contraction kernel generalizing (#3126)

2025-12-02 13:30:27 +01:00

batched_transpose

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

common

[CK_Tile] Support for a4w4 (fp4) in block scale gemm AB quant (#3603)

2026-01-30 04:40:50 -07:00

elementwise

[CK_Tile] Support for a4w4 (fp4) in block scale gemm AB quant (#3603)

2026-01-30 04:40:50 -07:00

epilogue

fix mxfp8-gemm example failure (#3531)

2026-01-13 10:26:45 +08:00

flatmm

Solve the CTAD regression & add up the Shell file for the docker management in testing (#3634)

2026-01-26 10:29:28 -08:00

fmha

Optimize batch prefill kernel performance for VECTORIZED_LAYOUT KV cache (#3657)

2026-01-29 07:18:41 +08:00

fused_moe

Shuffle fix for gfx950 (#3491)

2026-01-13 09:21:29 -08:00

gemm

[CK_Tile] Support for a4w4 (fp4) in block scale gemm AB quant (#3603)

2026-01-30 04:40:50 -07:00

gemm_quant

[CK_Tile] Support for a4w4 (fp4) in block scale gemm AB quant (#3603)

2026-01-30 04:40:50 -07:00

grouped_convolution

[CK_BUILDER] Add grouped conv fwd ck tile profiler (#3518)

2026-01-19 22:29:01 -07:00

image_to_column

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

layernorm2d

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

norm_reduce

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

permute

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

pooling

Shuffle fix for gfx950 (#3491)

2026-01-13 09:21:29 -08:00

reduce

[CK Tile] multi reduce improvements (#3607)

2026-01-27 12:56:09 -08:00

rmsnorm2d

Fix redundant cast in model sensitive rmsnorm (#3681)

2026-01-30 10:52:19 +08:00

smoothquant

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

softmax/block

Solve the CTAD regression & add up the Shell file for the docker management in testing (#3634)

2026-01-26 10:29:28 -08:00

sparse_attn

[CK_TILE][FMHA] Add sparse attention VSA (#3341)

2026-01-31 00:59:47 +08:00

topk/block

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

topk_softmax

Shuffle fix for gfx950 (#3491)

2026-01-13 09:21:29 -08:00

add_rmsnorm2d_rdquant.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

batched_contraction.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

batched_transpose.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

common.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

elementwise.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

epilogue.hpp

[CK_TILE] Epilogue chaining (Lwpck 3373) (#2773)

2025-12-18 10:02:02 +01:00

flatmm.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

fmha.hpp

[FMHA] Batch Prefill Support Improvements: Change KV Cache Layout & Large Page Size Support (#3442)

2026-01-05 18:41:47 +08:00

fused_moe.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

gemm_quant.hpp

[CK_TILE] add preshuffleB mode for ABQuant GEMM (#3495)

2026-01-06 12:35:01 -08:00

gemm.hpp

Joye/revise wp pipeline (#3493)

2026-01-05 13:49:26 -08:00

grouped_convolution.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

image_to_column.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

layernorm2d.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

moe_flatmm.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

norm_reduce.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

permute.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

pooling.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

reduce.hpp

Dlejeune/ck tile 2d multiple reductions (#3147)

2026-01-09 11:16:37 +01:00

rmsnorm2d.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

smoothquant.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

softmax.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

sparse_attn.hpp

[CK_TILE][FMHA] Add sparse attention VSA (#3341)

2026-01-31 00:59:47 +08:00

topk_softmax.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00

topk.hpp

feat(precommit-hooks): add check for correct copyright header (#3302)

2025-12-10 22:50:43 -08:00