mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 05:31:24 +00:00

Files

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

* Finished the feature of gpu verification

* Add the ck_tile_gemm test in the CI CD

* add the include of tensor_layou in reference_gemm

* Comment Addressed

* split ck_tile fhma and gemm tests into separate stages

* restructure the reference gemm

* restructure a new reference_gemm api that could read the device mem

---------

Co-authored-by: carlushuang <carlus.huang@amd.com>
Co-authored-by: illsilin <Illia.Silin@amd.com>

2024-09-14 21:08:40 +08: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 GPU verification sample develop & Add the CK TILE GEMM to the CI/CD test (#1505 )

2024-09-14 21:08:40 +08: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)