Skip to content

Commit e0cf5fe

Browse files
authored
[README] Update GDN README for clarity and add acknowledgements (#758)
- Improved formatting and clarity of the GDN kernel implementation description. - Updated requirement section to list dependencies in a clearer format. - Added an acknowledgements section to credit the developers and the Xiaomi LLM-Core Team for their contributions.
1 parent 556d411 commit e0cf5fe

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

examples/gdn/README.md

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,14 @@
1-
# Gated Delta Net(GDN) kernel implementation in TileLang
1+
# Gated Delta Net (GDN) kernel implementation with TileLang
22

33
## Requirement
44

5-
### The Tilelang version for test is 0.1.5+17fafc1b3026d910a83eb8052fdf811ba56be0b1
6-
7-
### We currently use triton=3.3.0 and FLA commit id=f03cb3ae for comparison
5+
- TileLang: `0.1.5+17fafc1b3026d910a83eb8052fdf811ba56be0b1`
6+
- Triton: `3.3.0` (used for comparison)
7+
- FLA: commit `f03cb3ae` (used for comparison)
88

99
## Get started
1010

11-
### The common/chunk_delta_h.py implements the most critical forward kernel of GDN. It's a good start to understand the GDN logic and the tilelang optimization
11+
The [chunk_delta_h](common/chunk_delta_h.py) implements the most critical forward kernel of GDN. It's a good start to understand the GDN logic and the TileLang optimization.
12+
13+
## Acknowledgements
14+
This kernel was developed by Yu Cheng and Zhengju Tang following in-depth discussions with Xiaomi's LLM-Core Team (MiMo).

0 commit comments

Comments
 (0)