composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00

Files

arai713 71986ea795 [CK TILE] Stream-K tile partitioner (#2708 )

* initial commit for skeleton code

* replaced skeleton code with old streamk b2c map functions from old CK, still need to clean up the code

* fixed up code to match CK Tile convention: data type changes, naming changes, etc.

* change for num_sk_blocks data type

* formatting fix

* minor fixes

* moved reduction argument to template

* resolved comments from PR review: standardizing naming, pruning unneeded code

* resolve errors from merge of device op PR: moved enum to common file

* switching to uint32_t due to implementation constraints: divmod only takes uint32_t and mixing signed and unsigned types causes problems

* unsigned type fix

* add const qualifier

* added documentation for template parameters

* documentation edit

[ROCm/composable_kernel commit: 0282d98412]

2025-09-03 13:38:17 -07:00

generic_2d_block_shape.hpp

[Ck tile] layernorm2d fwd optimize (#1637 )

2024-11-08 12:28:23 +08:00

README.md

introducing ck_tile! (#1216 )

2024-04-15 19:27:12 -05:00

streamk_common.hpp

[CK TILE] Stream-K tile partitioner (#2708 )

2025-09-03 13:38:17 -07:00

tensor_layout.hpp

introducing ck_tile! (#1216 )

2024-04-15 19:27:12 -05:00

utils.hpp

chore: unset executable permission (#2303 )

2025-06-10 09:13:59 -07:00

README.md

common

this folder is designed not to be included directly by use, e.g. if use include ck_tile/ops/fmha.hpp, then everything under common should also be included.

to achieve this we will duplicate the header include path under common to other module under ops/* inside remod.py. for internal developer, you can also include ck_tile/ops/common.hpp for convenience. (and so does external users...)