Papers by Yangjie Zhou
CuBridge: An LLM-Based Framework for Understanding and Reconstructing High-Performance Attention Kernels (2026.acl-long)
Copied to clipboard
Xing Ma, Yangjie Zhou, Wu Sun, Zihan Liu, Jingwen Leng, Yun Lin, Shixuan Sun, Minyi Guo, Jin Song Dong
| Challenge: | Existing approaches to support diverse attention variants trade performance for flexibility . expert-written kernels achieve high efficiency but are difficult to adapt . |
| Approach: | They propose a framework that adapts expert-written attention kernels to GPUs . they use a structured lift–transfer–lower workflow to make execution explicit . |
| Outcome: | The proposed framework outperforms existing frameworks and compilers on diverse variants and GPU platforms. |