mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-05 14:11:29 +00:00

Files

kylasa ab5d027866 Support for dtypes (fp8, bf8, bf16 and fp16) for the ck_tile/03_gemm example. (#1845 )

* Support bf16/fb8/bf8 datatypes for ck_tile/gemm

* remove commented out code.

* Addressing code review comments and enabling universal_gemm for all the supported data types.

* Merge conflict resolution.

* Solve the memory pipeline compilation error. Merge with the new change of CShuffle

* finish the feature, pass the tests

* Fix the pipeline and add the benchmark script for other data types

---------

Co-authored-by: ThomasNing <thomas.ning@amd.com>

2025-02-07 15:07:06 -07:00

algorithm

[CK-Tile] Enable vectorized reads on all layouts & improve perf. (#1835 )

2025-01-27 16:37:19 +01:00

arch

Support for dtypes (fp8, bf8, bf16 and fp16) for the ck_tile/03_gemm example. (#1845 )

2025-02-07 15:07:06 -07:00

container

[CK-Tile] Enable vectorized reads on all layouts & improve perf. (#1835 )

2025-01-27 16:37:19 +01:00

numeric

Fix duplication of pk_add_f16 symbols (#1858 )

2025-02-04 14:42:11 +01:00

tensor

[CK-Tile] Enable vectorized reads on all layouts & improve perf. (#1835 )

2025-01-27 16:37:19 +01:00

utility

add fp8 as dst (#1830 )

2025-01-22 17:34:27 +08:00

config.hpp

Fix pk_int4 cast and add pk_int4 dtype in ck tile (#1854 )

2025-02-04 10:32:07 +01: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