mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 13:41:24 +00:00

Files

damien-lejeune 4216d43da8 Dlejeune/ck tile 2d multiple reductions (#3147 )

* WIP

* Add Unit tests for the Multi Reduction Kernel

* clang format

* Rename multiblock to threadwise

* Multiblock WIP

* Fix multi reduce multi block unit tests

* Multi Reduce Tile Engine: WIP

* refactoring + try addressing precision error

* Fix multiops examples

* Cleanup

* Clean up tile engine's reduce op

* Update changelog

* Fix remod/clang

* Fix dates

* Fix documentation & missing file

* Fix comments

* Use the update_tile api in the multi-block kernel

* Unify threadwise/multiblock into a single kernel + default multiblock output to float in tests

* Add TileParitioner

* Cleanup

* Add warning when no data to process, in the example

* Refactoring Reduce kernel Tile Partioner + cleanup

* Move the tile partioner to its own file

* Add missing includes

* Fix copyright header with update_amd_copyright_headers.py

* Fix change of interface in Reduce2dProblem

---------

Co-authored-by: Damien Lejeune <damien.lejeune@amd.com>
Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>

2026-01-09 11:16:37 +01:00

algorithm

[CK_TILE][FMHA] Integrate FAv2 & FAv3 (WIP) in the single fmha_fwd() API (#3153 )

2025-12-05 10:31:12 +08:00

arch

[CK TILE] Refactor function amd_buffer_load_invalid_element_return_zero (#3512 )

2026-01-07 00:05:56 -08:00

container

chore(copyright): update copyright header for include directory (#3293 )

2025-11-26 11:00:05 -07:00

numeric

flashattention fwd add (80, 96) instance (#3415 )

2025-12-17 09:16:11 -08:00

tensor

Merge some updates for ck_tile headers (#3342 )

2026-01-05 23:39:00 -08:00

utility

Dlejeune/ck tile 2d multiple reductions (#3147 )

2026-01-09 11:16:37 +01:00

config.hpp

Add support to gfx1153 and fix gfx115X WMMA config (#3496 )

2026-01-05 10:03:30 -08:00

README.md

introducing ck_tile! (#1216 )

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

README.md

ck_tile/core

ck_tile/core contains every basic functions and structures to create a GPU kernel using ck_tile. User should only include ck_tile/core.hpp this single header to use all the functionality. Everything is under ck_tile namespace. The coding style under this folder should be similar to std (snake_case for structure/function, Camel for template types...)

algorithm/
    coordinate transform and some other reusable algorithm
arch/
    contains some basic device building block like mma, buffer addressing, etc...
container/
    contains basic container data structure, array/sequence/tuple/...
numeric/
    data type, and data type related math
tensor/
    tensor descriptors and tile level API
utility/
    other utility function for both host/device