composable_kernel/example/ck_tile/99_toy_example

CK_TILE Toy Example

This repository demonstrates a toy example implemented using ck_tile

Build Instructions

Follow these steps to build the examples:

cd composable_kernel
mkdir build
cd build

cmake -D CMAKE_PREFIX_PATH=/opt/rocm \
      -D CMAKE_CXX_COMPILER=/opt/rocm/bin/hipcc \
      -D CMAKE_BUILD_TYPE=Release \
      -D GPU_TARGETS="gfx942" \
      -Dkernel=N ..

Compile Examples

Elementwise Add Example

make -j add

GEMM Example

make -j basic_gemm

Flash Attention Forward Example

make -j basic_flash_attention_fwd

Running Examples

Elementwise Add

./bin/add

GEMM Example

./bin/basic_gemm 1

Flash Attention Forward Example

./bin/basic_flash_attention_fwd 1 0 1