paperbot · PL 论文追踪

全部论文 待读 收藏

P1 Data-driven Test Generation for Fuzzing AI Compiler

面向 AI 编译器模糊测试的数据驱动测试生成
ICSE Proceedings of the 2026 IEEE/ACM 48th International Conference on Software Engineering2026匹配度 5/10 · 论文关注 AI 编译器测试生成(fuzzing),属于机器学习系统质量保障,与用户“在 ML 系统/编译器中应用程序分析”有领域交叉;但核心方法是数据驱动测试而非静态分析,且未涉及 LLM/Agent、浮点程序分析或贝叶斯程序分析,因此匹配度中等。引用 0 inbox
Qingchao Shen
crossref: 2026-09-12

AI 速览

一句话:提出统一数据驱动测试框架,面向 AI 编译器各阶段生成测试,发现 266 个未知缺陷。

AI 编译器对将模型高效部署到多样硬件至关重要,但其自身缺陷会危及编译器可靠性与模型正确性。本文提出一个统一的数据驱动测试框架,系统应对 AI 编译器不同阶段的测试挑战:OPERA 将 AI 库测试迁移用于模型加载阶段的算子转换逻辑测试;OATest 合成多样化且优化感知的计算图,以测试高层优化;HARMONY 生成并变异多样的低层 IR 种子,以生成硬件优化感知的测试。三者结合形成阶段感知的综合测试框架,提升了测试覆盖率和有效性,在四个广泛使用的 AI 编译器中检测到 266 个此前未知的缺陷。

问题
AI 编译器在不同阶段均易引入缺陷,威胁编译器可靠性与模型正确性,而现有测试方法难以系统覆盖其阶段特定逻辑。
方法
提出统一数据驱动测试框架,包含 OPERA、OATest 与 HARMONY 三项技术,分别针对模型加载、高层优化与低层优化阶段生成测试。
评估
在四个广泛使用的 AI 编译器上进行评估,共检测到 266 个此前未知的缺陷,表明该框架能提升测试覆盖率与有效性。
原文摘要(Abstract)

Artificial Intelligence (AI) compilers are critical for efficiently deploying AI models across diverse hardware platforms. However, they remain prone to bugs that can compromise both compiler reliability and model correctness. Thus, ensuring the quality of AI compilers is crucial. In this work, we present a unified data-driven testing framework that systematically addresses stage-specific challenges in AI compilers. Specifically, OPERA migrates tests for AI libraries to test various operator conversion logic in the model loading stage. OATest synthesizes diverse optimization-aware computational graphs for testing high-level optimizations. HARMONY generates and mutates diverse low-level IR seeds to generate hardware-optimization-aware tests for testing low-level optimizations. Together, these techniques provide a comprehensive, stage-aware framework that enhances testing coverage and effectiveness, detecting 266 previously unknown bugs in four widely used AI compilers.

链接与引用

DOI 原文 · PDF(开放获取) ·

BibTeX
@inproceedings{paperbot11929,
  title = {Data-driven Test Generation for Fuzzing AI Compiler},
  author = {Qingchao Shen},
  booktitle = {ICSE},
  year = {2026},
  doi = {10.1145/3774748.3787621}
}