mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 15:54:31 +00:00

Files

yadaish c0ee71d735 Dev/a8w4 and a8w8splitk (#3447 )

* Ck moe bs splitk pr (#3440)

* splitk kick-off. Compilation fail

* splitk hack pass

* fix scale offset calc.

* clang-format for a8w8_moe_blk_gemm1 splitk change

* fix testcase error

---------

Co-authored-by: oscar <huaiguxu@amd.com>
Co-authored-by: huaiguxu <145733371+huaiguxu@users.noreply.github.com>

* Zan/moe a8w4 (#3441)

* update

* update

* update ck moe a8w4

* update

* update

* update

* compile pass

* update

* update

* python3 op_tests/test_moe_2stage.py -t 16 -e 1 -k 1 -dim 256,256 ready

* support new a8w4 kernel

* update

* update ck_tile

* re format

* update

* update

* fix conflict

* fix build

* update ck_tile moe

* fix clang format

* fix the problem

* fix accruacy issue

* fix

---------

Co-authored-by: oscar <huaiguxu@amd.com>
Co-authored-by: huaiguxu <145733371+huaiguxu@users.noreply.github.com>
Co-authored-by: Zzz9990 <zanzhang@amd.com>
Co-authored-by: felix <felix.li@amd.com>

2025-12-19 09:26:52 +08: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

Update AMD buffer coherency (#3403 )

2025-12-18 10:16:22 +01: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

Dev/a8w4 and a8w8splitk (#3447 )

2025-12-19 09:26:52 +08:00

utility

Fix compilation errors with latest clang22 version. (#3396 )

2025-12-11 08:09:29 -08:00

config.hpp

build: reduce build time for bquant tests by splitting into multiple cpp & support on other gfx10 case (#3395 )

2025-12-15 07:19:29 -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