ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 05:01:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
94d47b1680eaafacca142f2498fb94d08a5b66d3
composable_kernel/example/ck_tile/01_fmha/codegen
History
joyeamd 94d47b1680 fmha hdim256 vectorize improve (#2086) 
For hdim 256, will not have vectorized buffer load when seqlen % 256 != 0 and hdim % 256 = 0; this commit tries to solve this condition.
2025-04-16 09:21:04 +08:00
..
ops
fmha hdim256 vectorize improve (#2086)
2025-04-16 09:21:04 +08:00
__init__.py
[CK_TILE] fmha forward split-kv + combine kernels (#1338)
2024-06-26 17:41:15 +08:00
cmake_config.py
[CK_TILE] fmha forward split-kv + combine kernels (#1338)
2024-06-26 17:41:15 +08:00
cpp_symbol_map.py
[CK_TILE] Add fmha fwd N-Warp S-Shuffle pipeline (fmha fwd splitkv pipeline variant) (#1705)
2024-12-20 14:41:01 +08:00