Create README.md

This commit is contained in:
ClementLinCF
2025-04-01 09:11:29 +08:00
committed by GitHub
parent d1dbc69eda
commit 04513ca683

View File

@@ -0,0 +1,55 @@
# CK_TILE Toy Example
This repository demonstrates a toy example implemented using ck_tile
## Build Instructions
Follow these steps to build the examples:
```sh
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**
```sh
make -j add
```
#### **GEMM Example**
```sh
make -j basic_gemm
```
#### **Flash Attention Forward Example**
```sh
make -j basic_flash_attention_fwd
```
## Running Examples
### **Elementwise Add**
```sh
./bin/add
```
### **GEMM Example**
```sh
./bin/basic_gemm 1
```
### **Flash Attention Forward Example**
```sh
./bin/basic_flash_attention_fwd 1 0 1
```