Compiler-Grounded Hierarchical Diagnosis for LLM-Based Triton Kernel Optimization
Huawei Technologies Co., Ltd.
arXiv:2607.23089 [cs.AI], (25 Jul 2026)
@misc{chen2026compilergrounded,
title={Compiler-Grounded Hierarchical Diagnosis for LLM-Based Triton Kernel Optimization},
author={Dongjie Chen and Ping Zhao and Bohua Zhan and Yulong Wang and Shushu Chen and Liangjun Feng and Hao Zhou and Min Shen and Linmu Wang and Weijia Sheng and Xiangyu Wei and Weijie Ding and Jianhui Huang and Yaoqing Gao},
year={2026},
eprint={2607.23089},
archivePrefix={arXiv},
primaryClass={cs.AI},
url={https://arxiv.org/abs/2607.23089}
}
Recent advances in large language models (LLMs) have enabled automated kernel generation and optimization, but most existing approaches rely on surface signals such as compilation feedback and profiling metrics. These signals reveal that a kernel is slow, but not why the backend compiler fails to realize a profitable optimization, especially on emerging accelerators such as NPUs. We therefore formulate kernel optimization as a progressive cross-layer diagnosis problem that links runtime symptoms to IR structure and compiler behavior before rewriting source. Based on this insight, we present our system, a compiler-grounded and hierarchical optimization framework for Triton kernels. the system escalates from lightweight pattern triage and profiling diagnosis to IR attribution and compiler-grounded analysis only when deeper evidence is needed, then proposes evidence-backed source-level rewrites. We implement the system on Triton for Ascend NPUs and evaluate it on 37 successfully converted entries from a standardized NPUKernelBench-derived Ascend 950 benchmark. Across these entries, the system attains a geometric-mean speedup of 4.35x and a median speedup of 2.73x from the initial to optimized Triton kernel; 22/37 exceed 2x and 13/37 exceed 5x. The complete distribution ranges from near-baseline entries to large wins, motivating transparent reporting of the current system’s scope and limitations.
August 4, 2026 by hgpu
Your response
You must be logged in to post a comment.





