mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 05:01:25 +00:00

Files

Thomas Ning 694c300145 Ck tile gemm padding dim (#1516 )

* Support the N dimension padding

* Finished the padding feature for different dimension of K

2024-09-18 11:32:29 -07:00

script

Ck tile GPU verification sample develop & Add the CK TILE GEMM to the CI/CD test (#1505 )

2024-09-14 21:08:40 +08:00

CMakeLists.txt

Ck tile gemm example (#1488 )

2024-09-07 16:23:32 +08:00

gemm_basic.cpp

Ck tile gemm padding dim (#1516 )

2024-09-18 11:32:29 -07:00

gemm_basic.hpp

Ck tile GPU verification sample develop & Add the CK TILE GEMM to the CI/CD test (#1505 )

2024-09-14 21:08:40 +08:00

README.md

Ck tile gemm example (#1488 )

2024-09-07 16:23:32 +08:00

README.md

GEMM Matrix Multiplication

This folder contains example for GEMM using ck_tile tile-programming implementation. Currently, it only supports the basic feature of the CK Tile GEMM, but creates the placeholders for the future support on different GEMM pipeline and different GEMM modules. In the near future, we will gradually migrate all the GEMM features from old CK to CK Tile.

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_example_gemm_basic -j

This will result in an executable build/bin/tile_example_gemm_basic

example

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