PWCET: Power-Aware Worst Case Execution Time Analysis. - Wenlei Bao; Sanket Tavarageri; Füsun Özgüner; P. Sadayappan (ICPP Workshops, 2014)

Effective padding of multidimensional arrays to avoid cache conflict misses. - Changwan Hong; Wenlei Bao; Albert Cohen 001; Sriram Krishnamoorthy; Louis-Noël Pouchet; Fabrice Rastello; J. Ramanujam; P. Sadayappan (PLDI, 2016)

PolyCheck: dynamic verification of iteration space transformations on affine programs. - Wenlei Bao; Sriram Krishnamoorthy; Louis-Noël Pouchet; Fabrice Rastello; P. Sadayappan (POPL, 2016)

Static and Dynamic Frequency Scaling on Multicore CPUs. - Wenlei Bao; Changwan Hong; Sudheer Chunduri; Sriram Krishnamoorthy; Louis-Noël Pouchet; Fabrice Rastello; P. Sadayappan (ACM Trans. Archit. Code Optim., 2016)

Efficient Cache Simulation for Affine Computations. - Wenlei Bao; Prashant Singh Rawat; Martin Kong; Sriram Krishnamoorthy; Louis-Noël Pouchet; P. Sadayappan (LCPC, 2017)

Analytical modeling of cache behavior for affine programs. - Wenlei Bao; Sriram Krishnamoorthy; Louis-Noël Pouchet; P. Sadayappan (Proc. ACM Program. Lang., 2018)

NGEMM: Optimizing GEMM for Deep Learning via Compiler-based Techniques. - Wenlei Bao; Li-Wen Chang; Yang Chen; Ke Deng; Amit Agarwal; Emad Barsoum; Abe Taha (CoRR, 2019)

FLUX: Fast Software-based Communication Overlap On GPUs Through Kernel Fusion. - Li-Wen Chang; Wenlei Bao; Qi Hou; Chengquan Jiang; Ningxin Zheng; Yinmin Zhong; Xuanrun Zhang; Zuquan Song; Ziheng Jiang; Haibin Lin; Xin Jin 008; Xin Liu 086 (CoRR, 2024)

ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference. - Hanshi Sun; Li-Wen Chang; Wenlei Bao; Size Zheng 001; Ningxin Zheng; Xin Liu 086; Harry Dong; Yuejie Chi; Beidi Chen (CoRR, 2024)

Comet: Fine-grained Computation-communication Overlapping for Mixture-of-Experts. - Shulai Zhang; Ningxin Zheng; Haibin Lin; Ziheng Jiang; Wenlei Bao; Chengquan Jiang; Qi Hou; Weihao Cui; Size Zheng 001; Li-Wen Chang; Quan Chen 002; Xin Liu 086 (CoRR, 2025)

TileLink: Generating Efficient Compute-Communication Overlapping Kernels using Tile-Centric Primitives. - Size Zheng 001; Jin Fang; Xuegui Zheng; Qi Hou; Wenlei Bao; Ningxin Zheng; Ziheng Jiang; Dongyang Wang; Jianxi Ye; Haibin Lin; Li-Wen Chang; Xin Liu 086 (CoRR, 2025)

Triton-distributed: Programming Overlapping Kernels on Distributed AI Systems with the Triton Compiler. - Size Zheng 001; Wenlei Bao; Qi Hou; Xuegui Zheng; Jin Fang; Chenhui Huang; Tianqi Li; Haojie Duanmu; Renze Chen; Ruifan Xu; Yifan Guo; Ningxin Zheng; et al. (CoRR, 2025)

MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production. - Chao Jin; Ziheng Jiang; Zhihao Bai; Zheng Zhong; Juncai Liu; Xiang Li 067; Ningxin Zheng; Xi Wang; Cong Xie; Qi Huang; Wen Heng; Yiyuan Ma; et al. (CoRR, 2025)