mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-15 18:42:06 +00:00

Files

Illia Silin 24f228df3b upgrade from clang-format-12 to clang-format-18 (#2568 )

* upgrade to clang-format-18

* update to clang-format-18 in pre-commit-config

[ROCm/composable_kernel commit: 504b101da3]

2025-07-28 11:34:07 -07:00

instances

[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862 )

2025-02-20 14:00:27 -08:00

script

[Ck_tile] smoothquant (#1617 )

2024-11-01 13:51:56 +08:00

add_rmsnorm2d_rdquant_fwd.cpp

upgrade from clang-format-12 to clang-format-18 (#2568 )

2025-07-28 11:34:07 -07:00

add_rmsnorm2d_rdquant_fwd.hpp

[CK_TILE] fix build error in tile_add_rmsnorm2d_rdquant_fwd (#2243 )

2025-06-17 21:37:59 -07:00

CMakeLists.txt

Label CMakeLists message() as DEBUG or STATUS for clean build output (#2301 )

2025-06-10 10:46:47 -07:00

example_add_rmsnorm2d_rdquant_fwd.cpp

upgrade from clang-format-12 to clang-format-18 (#2568 )

2025-07-28 11:34:07 -07:00

README.md

[Ck tile] support rmsnorm and related fusion (#1605 )

2024-10-30 15:22:56 +08:00

README.md

Add + Rmsnorm2D + rowwise dynamic quantization forward

This folder contains example for add + Rmsnorm2D + rowwise dynamic quantization forward using ck_tile tile-programming implementation. Rdquant is short for rowwise dynamic quantization here.

build

# in the root of ck_tile
mkdir build && cd build
sh ../script/cmake-ck-dev.sh  ../ <arch>  # you can replace this <arch> to gfx90a, gfx942...
make tile_add_rmsnorm2d_rdquant_fwd -j

This will result in an executable build/bin/tile_add_rmsnorm2d_rdquant_fwd

cmdline

args:
          -m    m dimension (default:3328)
          -n    m dimension (default:4096)
          -e    epsilon (default:1e-5)
          -v    cpu validation or not (default:1)
       -prec    precision (default:fp16)