Yuanbo Wen

According to our database, Yuanbo Wen authored at least 42 papers between 2020 and 2026.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of four.

2026

DASA: Distribution-Aware Sparse Attention for Accelerating Diffusion Transformer.

Tianyun Ma, Jiaming Guo, Di Huang, Xinkai Song, Qian Wang, Husheng Han, Pengwei Jin, Xiangtao Guan, Yifan Hao, Yuanbo Wen, Shuyao Cheng, Zidong Du, Rui Zhang, Xing Hu, Qi Guo

IEEE Trans. Comput. Aided Des. Integr. Circuits Syst., June, 2026

AGON: Automated Design Framework for Customizing Processors From ISA Documents.

Chongxiao Li, Di Huang, Pengwei Jin, Tianyun Ma, Husheng Han, Shuyao Cheng, Yifan Hao, Yongwei Zhao, Guanglin Xu, Zidong Du, Rui Zhang, Xiaqing Li, Yuanbo Wen, Xing Hu, Qi Guo

IEEE Trans. Comput. Aided Des. Integr. Circuits Syst., May, 2026

FlashAttention-T: Towards Fully Tensorized Attention by Exploiting Tensor-Vector Parallelism.

Jianxing Xu, Yuanbo Wen, Jun Bi, Ruibai Xu, Guanglin Xu, Rui Zhang, Wei Li, Ling Li, Tianshi Chen, Qi Guo, Yunji Chen

Proceedings of the 31st ACM SIGPLAN Annual Symposium on Principles and Practice of Parallel Programming, 2026

QiMeng-PRepair: Precise Code Repair via Edit-Aware Reward Optimization.

Changxin Ke, Rui Zhang, Jiaming Guo, Yuanbo Wen, Li Ding, Shuo Wang, Xuyuan Zhu, Xiong Peng, Di Huang, Zidong Du, Xing Hu, Qi Guo, Yunji Chen

Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), 2026

QiMeng-Kernel: Macro-Thinking Micro-Coding Paradigm for LLM-Based High-Performance GPU Kernel Generation.

Xinguo Zhu, Shaohui Peng, Jiaming Guo, Yunji Chen, Qi Guo, Yuanbo Wen, Hang Qin, Ruizhi Chen, Qirui Zhou, Ke Gao, Yanjun Wu, Chen Zhao, Ling Li

Proceedings of the Fortieth AAAI Conference on Artificial Intelligence, 2026

2025

VariPar: Variation-Aware Workload Partitioning in Chiplet-Based DNN Accelerators.

Ximing Liu, Yongwei Zhao, Mo Zou, Yang Liu, Yifan Hao, Xiaqing Li, Rui Zhang, Yuanbo Wen, Xing Hu, Zidong Du, Qi Guo, Tianshi Chen

IEEE Trans. Comput. Aided Des. Integr. Circuits Syst., December, 2025

Artifact for the PPoPP '26 paper: "FlashAttention-T: Towards Fully Tensorized Attention by Exploiting Tensor-Vector Parallelism.".

Jianxing Xu, Yuanbo Wen, Jun Bi, Ruibai Xu

Dataset, November, 2025

QiMeng-Attention: SOTA Attention Operator is generated by SOTA Attention Algorithm.

Qirui Zhou, Shaohui Peng, Weiqiang Xiong, Haixin Chen, Yuanbo Wen, Haochen Li, Ling Li, Qi Guo, Yongwei Zhao, Ke Gao, Ruizhi Chen, Yanjun Wu, Chen Zhao, Yunji Chen

CoRR, June, 2025

Mutual-Supervised Learning for Sequential-to-Parallel Code Translation.

Changxin Ke, Rui Zhang, Shuo Wang, Li Ding, Guangli Li, Yuanbo Wen, Shuoming Zhang, Ruiyuan Xu, Jin Qin, Jiaming Guo, Chenxi Wang, Ling Li, Qi Guo, Yunji Chen

CoRR, June, 2025

QiMeng: Fully Automated Hardware and Software Design for Processor Chip.

Rui Zhang, Yuanbo Wen, Shuyao Cheng, Di Huang, Shaohui Peng, Jiaming Guo, Pengwei Jin, Jiacheng Zhao, Tianrui Ma, Yaoyu Zhu, Yifan Hao, Yongwei Zhao, Shengwen Liang, Ying Wang, Xing Hu, Zidong Du, Huimin Cui, Ling Li, Qi Guo, Yunji Chen

CoRR, June, 2025

QiMeng-TensorOp: Automatically Generating High-Performance Tensor Operators with Hardware Primitives.

Xuzhi Zhang, Shaohui Peng, Qirui Zhou, Yuanbo Wen, Qi Guo, Ruizhi Chen, Xinguo Zhu, Weiqiang Xiong, Haixin Chen, Congying Ma, Ke Gao, Chen Zhao, Yanjun Wu, Yunji Chen, Ling Li

CoRR, May, 2025

QiMeng-CPU-v2: Automated Superscalar Processor Design by Learning Data Dependencies.

Shuyao Cheng, Rui Zhang, Wenkai He, Pengwei Jin, Chongxiao Li, Zidong Du, Xing Hu, Yifan Hao, Guanglin Xu, Yuanbo Wen, Ling Li, Qi Guo, Yunji Chen

CoRR, May, 2025

QiMeng-Xpiler: Transcompiling Tensor Programs for Deep Learning Systems with a Neural-Symbolic Approach.

Shouyang Dong, Yuanbo Wen, Jun Bi, Di Huang, Jiaming Guo, Jianxing Xu, Ruibai Xu, Xinkai Song, Yifan Hao, Xuehai Zhou, Tianshi Chen, Qi Guo, Yunji Chen

CoRR, May, 2025

MigGPT: Harnessing Large Language Models for Automated Migration of Out-of-Tree Linux Kernel Patches Across Versions.

Pucheng Dang, Di Huang, Dong Li, Kang Chen, Yuanbo Wen, Qi Guo, Xing Hu, Ninghui Sun

CoRR, April, 2025

Harmonia: A Unified Architecture for Efficient Deep Symbolic Regression.

Tianyun Ma, Yuanbo Wen, Xinkai Song, Pengwei Jin, Di Huang, Husheng Han, Ziyuan Nan, Zhongkai Yu, Shaohui Peng, Yongwei Zhao, Huaping Chen, Zidong Du, Xing Hu, Qi Guo

IEEE Trans. Comput. Aided Des. Integr. Circuits Syst., February, 2025

Efficient and Fast High-Performance Library Generation for Deep Learning Accelerators.

Jun Bi, Yuanbo Wen, Xiaqing Li, Yongwei Zhao, Yuxuan Guo, Enshuai Zhou, Xing Hu, Zidong Du, Ling Li, Huaping Chen, Tianshi Chen, Qi Guo

IEEE Trans. Computers, January, 2025

AI Computing Systems for Large Language Models Training.

Zhenxing Zhang, Yuanbo Wen, Han-Qi Lyu, Chang Liu, Rui Zhang, Xia-Qing Li, Chao Wang, Zidong Du, Qi Guo, Ling Li, Xue-Hai Zhou, Yun-Ji Chen

J. Comput. Sci. Technol., January, 2025

QiMeng-MuPa: Mutual-Supervised Learning for Sequential-to-Parallel Code Translation.

Changxin Ke, Rui Zhang, Shuo Wang, Li Ding, Guangli Li, Yuanbo Wen, Shuoming Zhang, Ruiyuan Xu, Jin Qin, Jiaming Guo, Chenxi Wang, Ling Li, Qi Guo, Yunji Chen

Proceedings of the Advances in Neural Information Processing Systems 38: Annual Conference on Neural Information Processing Systems 2025, 2025

QiMeng-NeuComBack: Self-Evolving Translation from IR to Assembly Code.

Hainan Fang, Yuanbo Wen, Jun Bi, Yihan Wang, Tonghui He, Yanlin Tang, Di Huang, Jiaming Guo, Rui Zhang, Qi Guo, Yunji Chen

Proceedings of the Advances in Neural Information Processing Systems 38: Annual Conference on Neural Information Processing Systems 2025, 2025

Cambricon-SR: An Accelerator for Neural Scene Representation with Sparse Encoding Table.

Tianbo Liu, Xinkai Song, Zhifei Yue, Rui Wen, Xing Hu, Zhuoran Song, Yuanbo Wen, Yifan Hao, Wei Li, Zidong Du, Rui Zhang, Jiaming Guo, Di Huang, Shaohui Peng, Guangzhong Sun, Qi Guo, Tianshi Chen

Proceedings of the 52nd Annual International Symposium on Computer Architecture, 2025

QiMeng-TensorOp: One-Line Prompt is Enough for High-Performance Tensor Operator Generation with Hardware Primitives.

Xuzhi Zhang, Shaohui Peng, Qirui Zhou, Yuanbo Wen, Qi Guo, Ruizhi Chen, Xinguo Zhu, Weiqiang Xiong, Haixin Chen, Congying Ma, Ke Gao, Chen Zhao, Yanjun Wu, Yunji Chen, Ling Li

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence, 2025

Automated Superscalar Processor Design by Learning Data Dependencies.

Shuyao Cheng, Rui Zhang, Wenkai He, Pengwei Jin, Chongxiao Li, Zidong Du, Xing Hu, Yifan Hao, Guanglin Xu, Yuanbo Wen, Ling Li, Qi Guo, Yunji Chen

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence, 2025

Mosaic: Exploiting Instruction-Level Parallelism on Deep Learning Accelerators with <i>iTex</i> Tessellation.

Jianxing Xu, Yuanbo Wen, Zikang Liu, Ruibai Xu, Tingfeng Ruan, Jun Bi, Rui Zhang, Di Huang, Xinkai Song, Yifan Hao, Xing Hu, Zidong Du, Chongqing Zhao, Jiang Jie, Qi Guo

Proceedings of the 30th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, 2025

QiMeng-Attention: SOTA Attention Operator is generated by SOTA Attention Algorithm.

Qirui Zhou, Shaohui Peng, Weiqiang Xiong, Haixin Chen, Yuanbo Wen, Haochen Li, Ling Li, Qi Guo, Yongwei Zhao, Ke Gao, Ruizhi Chen, Yanjun Wu, Zhao Chen, Yunji Chen

Proceedings of the Findings of the Association for Computational Linguistics, 2025

QiMeng-GEMM: Automatically Generating High-Performance Matrix Multiplication Code by Exploiting Large Language Models.

Qirui Zhou, Yuanbo Wen, Ruizhi Chen, Ke Gao, Weiqiang Xiong, Ling Li, Qi Guo, Yanjun Wu, Yunji Chen

Proceedings of the Thirty-Ninth AAAI Conference on Artificial Intelligence, 2025

2024

AGON: Automated Design Framework for Customizing Processors from ISA Documents.

Chongxiao Li, Di Huang, Pengwei Jin, Tianyun Ma, Husheng Han, Shuyao Cheng, Yifan Hao, Yongwei Zhao, Guanglin Xu, Zidong Du, Rui Zhang, Xiaqing Li, Yuanbo Wen, Yanjun Wu, Chen Zhao, Xing Hu, Qi Guo

CoRR, 2024

Assessing and Understanding Creativity in Large Language Models.

Yunpu Zhao, Rui Zhang, Wenyi Li, Di Huang, Jiaming Guo, Shaohui Peng, Yifan Hao, Yuanbo Wen, Xing Hu, Zidong Du, Qi Guo, Ling Li, Yunji Chen

CoRR, 2024

Cambricon-C: Efficient 4-Bit Matrix Unit via Primitivization.

Yi Chen, Yongwei Zhao, Yifan Hao, Yuanbo Wen, Yuntao Dai, Xiaqing Li, Yang Liu, Rui Zhang, Mo Zou, Xinkai Song, Xing Hu, Zidong Du, Huaping Chen, Qi Guo, Tianshi Chen

Proceedings of the 57th IEEE/ACM International Symposium on Microarchitecture, 2024

Cambricon-D: Full-Network Differential Acceleration for Diffusion Models.

Weihao Kong, Yifan Hao, Qi Guo, Yongwei Zhao, Xinkai Song, Xiaqing Li, Mo Zou, Zidong Du, Rui Zhang, Chang Liu, Yuanbo Wen, Pengwei Jin, Xing Hu, Wei Li, Zhiwei Xu, Tianshi Chen

Proceedings of the 51st ACM/IEEE Annual International Symposium on Computer Architecture, 2024

Prompt-based Visual Alignment for Zero-shot Policy Transfer.

Haihan Gao, Rui Zhang, Qi Yi, Hantao Yao, Haochen Li, Jiaming Guo, Shaohui Peng, Yunkai Gao, QiCheng Wang, Xing Hu, Yuanbo Wen, Zihao Zhang, Zidong Du, Ling Li, Qi Guo, Yunji Chen

Proceedings of the Forty-first International Conference on Machine Learning, 2024

AutoOS: Make Your OS More Powerful by Exploiting Large Language Models.

Huilai Chen, Yuanbo Wen, Limin Cheng, Shouxu Kuang, Yumeng Liu, Weijia Li, Ling Li, Rui Zhang, Xinkai Song, Wei Li, Qi Guo, Yunji Chen

Proceedings of the Forty-first International Conference on Machine Learning, 2024

Revisiting Automatic Pipelining: Gate-level Forwarding and Speculation.

Shuyao Cheng, Chongxiao Li, Zidong Du, Rui Zhang, Xing Hu, Xiaqing Li, Guanglin Xu, Yuanbo Wen, Qi Guo

Proceedings of the 61st ACM/IEEE Design Automation Conference, 2024

TensorTEE: Unifying Heterogeneous TEE Granularity for Efficient Secure Collaborative Tensor Computing.

Husheng Han, Xinyao Zheng, Yuanbo Wen, Yifan Hao, Erhu Feng, Ling Liang, Jianan Mu, Xiaqing Li, Tianyun Ma, Pengwei Jin, Xinkai Song, Zidong Du, Qi Guo, Xing Hu

Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, 2024

2023

ANPL: Compiling Natural Programs with Interactive Decomposition.

Di Huang, Ziyuan Nan, Xing Hu, Pengwei Jin, Shaohui Peng, Yuanbo Wen, Rui Zhang, Zidong Du, Qi Guo, Yewen Pu, Yunji Chen

CoRR, 2023

ANPL: Towards Natural Programming with Interactive Decomposition.

Di Huang, Ziyuan Nan, Xing Hu, Pengwei Jin, Shaohui Peng, Yuanbo Wen, Rui Zhang, Zidong Du, Qi Guo, Yewen Pu, Yunji Chen

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

Emergent Communication for Rules Reasoning.

Yuxuan Guo, Yifan Hao, Rui Zhang, Enshuai Zhou, Zidong Du, Xishan Zhang, Xinkai Song, Yuanbo Wen, Yongwei Zhao, Xuehai Zhou, Jiaming Guo, Qi Yi, Shaohui Peng, Di Huang, Ruizhi Chen, Qi Guo, Yunji Chen

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

Cambricon-R: A Fully Fused Accelerator for Real-Time Learning of Neural Scene Representation.

Xinkai Song, Yuanbo Wen, Xing Hu, Tianbo Liu, Haoxuan Zhou, Husheng Han, Tian Zhi, Zidong Du, Wei Li, Rui Zhang, Chen Zhang, Lin Gao, Qi Guo, Tianshi Chen

Proceedings of the 56th Annual IEEE/ACM International Symposium on Microarchitecture, 2023

BALTO: fast tensor program optimization with diversity-based active learning.

Jun Bi, Xiaqing Li, Qi Guo, Rui Zhang, Yuanbo Wen, Xing Hu, Zidong Du, Xinkai Song, Yifan Hao, Yunji Chen

Proceedings of the Eleventh International Conference on Learning Representations, 2023

Heron: Automatically Constrained High-Performance Library Generation for Deep Learning Accelerators.

Jun Bi, Qi Guo, Xiaqing Li, Yongwei Zhao, Yuanbo Wen, Yuxuan Guo, Enshuai Zhou, Xing Hu, Zidong Du, Ling Li, Huaping Chen, Tianshi Chen

Proceedings of the 28th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, 2023

2022

Enabling One-Size-Fits-All Compilation Optimization for Inference Across Machine Learning Computers.

Yuanbo Wen, Qi Guo, Zidong Du, Jianxing Xu, Zhenxing Zhang, Xing Hu, Wei Li, Rui Zhang, Chao Wang, Xuehai Zhou, Tianshi Chen

IEEE Trans. Computers, 2022

BabelTower: Learning to Auto-parallelized Program Translation.

Yuanbo Wen, Qi Guo, Qiang Fu, Xiaqing Li, Jianxing Xu, Yanlin Tang, Yongwei Zhao, Xing Hu, Zidong Du, Ling Li, Chao Wang, Xuehai Zhou, Yunji Chen

Proceedings of the International Conference on Machine Learning, 2022

2020

Addressing Irregularity in Sparse Neural Networks Through a Cooperative Software/Hardware Approach.

Xi Zeng, Tian Zhi, Xuda Zhou, Zidong Du, Qi Guo, Shaoli Liu, Bingrui Wang, Yuanbo Wen, Chao Wang, Xuehai Zhou, Ling Li, Tianshi Chen, Ninghui Sun, Yunji Chen

IEEE Trans. Computers, 2020