jiangyon.ren 4d2f8c111e [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>

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

..

fmha_fwd_jenga_kernel.hpp

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

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

fmha_fwd_vsa_kernel.hpp

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

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