ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-11 17:51:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update example/ck_tile/42_unified_attention/README.md

Browse Source
This commit is contained in:
Tianxing Wu
2026-01-02 14:29:33 +02:00
committed by GitHub
parent a1973b472f
commit 74297016ea
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
example/ck_tile/42_unified_attention/README.md
View File

@@ -13,6 +13,7 @@ make tile_example_unified_attention -j
This will result in an executable `build/bin/tile_example_unified_attention`
## kernel
The kernel template is `unified_attention.hpp`, this is the grid-wise op in old ck_tile's terminology. We put it here purposely, to demonstrate one can construct a kernel by using various internal component from ck_tile. We may still have an implementation under ck_tile's include path (in the future) for the kernel template.
There are 2 template parameters for this kernel template.