mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-02 21:27:45 +00:00
Create README.md
This commit is contained in:
55
example/ck_tile/99_toy_example/README.md
Normal file
55
example/ck_tile/99_toy_example/README.md
Normal 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
|
||||
```
|
||||
Reference in New Issue
Block a user