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 |
|