mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 11:16:59 +00:00

Files

Clement Lin 35de33c57b Add codegen instances

The following examples have been tested for 04_codegen:

./bin/codegen_basic_flash_attention_fwd 1 1 64 4096 4096 256 256
./bin/codegen_basic_flash_attention_fwd 1 1 64 4096 4096 64 64
./bin/codegen_basic_flash_attention_fwd 1 1 64 4096 4096 32 32
./bin/codegen_basic_flash_attention_fwd 1 1 64 4096 4096 128 128
./bin/codegen_basic_flash_attention_fwd 1 1 64 2048 2048 128 128
./bin/codegen_basic_flash_attention_fwd 1 1 64 512 512 128 128

2025-04-23 11:51:35 +08:00

01_add

Fix error after clang-format

2025-04-22 15:24:55 +00:00

02_gemm

Fix error after clang-format

2025-04-22 15:24:55 +00:00

03_flash_attention_fwd

Add codegen instances

2025-04-23 11:51:35 +08:00

04_codegen_flash_attention_fwd

Add codegen instances

2025-04-23 11:51:35 +08:00

CMakeLists.txt

Add codegen test example

2025-04-10 22:30:59 +08:00

README.md

Create README.md

2025-04-01 09:11:29 +08:00

README.md

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