mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-01 12:11:19 +00:00

Files

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

* [CK_TILE] fix build error in tile_add_rmsnorm2d_rdquant_fwd

* fix error with the latest develop code.

2025-06-17 21:37:59 -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

…

add_rmsnorm2d_rdquant_fwd.cpp

Fix 11_add_rmsnorm2d_rdquant (#2207 )

2025-05-20 15:15:28 -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

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

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

README.md

…

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)