Unverified Commit e0cf5fee authored by Yu Cheng's avatar Yu Cheng Committed by GitHub
Browse files

[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.
parent 556d411e
# Gated Delta Net(GDN) kernel implementation in TileLang # Gated Delta Net (GDN) kernel implementation with TileLang
## Requirement ## Requirement
### The Tilelang version for test is 0.1.5+17fafc1b3026d910a83eb8052fdf811ba56be0b1 - TileLang: `0.1.5+17fafc1b3026d910a83eb8052fdf811ba56be0b1`
- Triton: `3.3.0` (used for comparison)
### We currently use triton=3.3.0 and FLA commit id=f03cb3ae for comparison - FLA: commit `f03cb3ae` (used for comparison)
## Get started ## Get started
### 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 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.
\ No newline at end of file
## Acknowledgements
This kernel was developed by Yu Cheng and Zhengju Tang following in-depth discussions with Xiaomi's LLM-Core Team (MiMo).
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment