← 最新论文
💻 computer science

A3D: Agentic AI flow for autonomous Accelerator Design

本文提出了 A3D,一种代理式人工智能框架,它能够自主编排工作负载分析、代码重构和微架构生成,从而在无需人工干预的情况下,为复杂的科学应用生成端到端的硬件加速器设计。

原作者: Abinand Nallathambi, Christopher Knight, Shantanu Ganguly, Wilfried Haensch, Anand Raghunathan

发布于 2026-05-18
📖 1 分钟阅读☕ 轻松阅读

原作者: Abinand Nallathambi, Christopher Knight, Shantanu Ganguly, Wilfried Haensch, Anand Raghunathan

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

想象一下,你拥有一份极其复杂、杂乱无章的 gourmet 大餐食谱(即科学计算机程序)。你想要打造一台特殊的定制厨房电器(即硬件加速器),使其烹饪这道特定菜肴的速度比标准炉灶快 1000 倍,且能耗更低。

问题在于?建造这台定制电器,就像蒙着眼睛去打造法拉利引擎。这需要一位既是主厨,又是机械大师、电气工程师,并且精通一种非常特定且艰深的语言——“硬件设计”的奇才。大多数编写食谱的人(科学家)并不具备这些技能,因此他们只能继续使用标准炉灶(通用计算机),而这些炉灶在处理这些特定任务时既缓慢又浪费。

A3D 登场:这支“超级主厨机器人”团队。

本文介绍了 A3D,这是一个新系统,它利用由 AI“智能体”(智能软件工人)组成的团队,从头到尾完成定制厨房电器的设计工作,无需人类动手。

以下是 A3D 的工作原理,分为三个简单阶段:

第一阶段:侦探工作(分析)

在开始构建之前,你需要确切知道要构建什么。

  • 问题: 科学程序规模庞大,包含数千行代码。找到那个拖慢整体速度的特定部分(即“瓶颈”),就像在 haystack 中找一根针,而这里 haystack 本身就是由无数根针组成的。
  • A3D 的解决方案: A3D 派出一支侦探团队。
    • 一个智能体阅读整份食谱,以理解其脉络。
    • 另一个智能体在测试厨房中运行食谱,对每一步进行计时,精准定位厨师浪费时间的环节。
    • 第三个智能体使用放大镜(静态分析),精确追踪该缓慢步骤所依赖的每一个食材和工具。
  • 结果: 他们精准定位了确切的“缓慢步骤”,并绘制出所有依赖关系,确保他们确切知道需要修复什么。

第二阶段:翻新(准备)

既然找到了缓慢步骤,就需要为定制机器对其进行准备。

  • 问题: 原始代码是用灵活但杂乱的语言(C++ 或 CUDA)编写的,而定制机器建造者(HLS 工具)无法理解。这就像试图将用古代象形文字绘制的蓝图交给现代建筑机器人。代码可能使用“动态内存”(例如抓取一个未知大小的盒子)或复杂的指针,而这些是机器人所厌恶的。
  • A3D 的解决方案: 一支翻新团队进驻。
    • 快照工程师: 对进入和离开缓慢步骤的数据进行“拍照”,以确保不会丢失食谱的风味。
    • 测试 harness 开发者: 搭建一个小型的隔离测试厨房,确保翻新不会破坏风味。
    • HLS 准备员: 这是主力。它将杂乱的代码重写为机器建造者能理解的干净、僵化的格式。它将复杂的工具替换为简单的工具,将“未知大小的盒子”转换为固定大小的板条箱,并将机器不认识的复杂数学函数替换为机器熟悉的函数。
    • 验证员: 一位严格的检查员审查翻新成果。如果代码不完美,他们将其退回给翻新员进行修复。他们不断循环,直到代码 100% 准备就绪。

第三阶段:构建与优化(综合)

现在代码已准备好被转化为硬件。

  • 问题: 构建机器有数百万种方式。你可以让它超快但体积巨大,或者体积小巧但速度慢。如何找到完美的平衡点?
  • A3D 的解决方案:
    • 数值探索者: 尝试计算中使用的数字的不同“精度”。也许你并不需要所有地方都使用 64 位精度;也许 20 位就足够了?这可以节省空间和功耗。
    • 综合器: 将清理后的代码交给商用机器建造者(Catapult HLS),以生成实际的硬件设计。
    • 设计空间枚举器: 这是总规划师。它运行数千次模拟,尝试速度与大小设置的所有可能组合。这就像测试汽车的所有可能齿轮比。
    • 结果: 它生成一份“帕累托最优”设计列表。这些是最佳的可能权衡:“这是最快版本”,“这是最小版本”,以及“这是最佳平衡版本”。

为什么这很重要?

以前的 AI 工具只能协助此过程中的小部分,例如修复代码中的拼写错误或建议一个设置。它们假设人类已经完成了寻找问题和清理代码的艰苦工作。

A3D 是首个完成整个工作的系统。 它处理了现实世界中庞大的科学程序(例如模拟分子或量子化学),并在零人类协助的情况下,将它们转化为定制硬件设计。

“秘密配方”(它为何真正有效)

论文测试了 A3D,发现它之所以有效,仅归功于三个特定技巧:

  1. 专业化: 不是让一个 AI 试图包揽所有事情(这会导致错误),而是使用一个专家团队,每个人完美地完成一项工作。
  2. 工具优于猜测: 当 AI 需要更改代码时,它不会只是“猜测”新文本。它使用精确的数学工具(例如代码结构的拼写检查器)来确保更改绝对正确。
  3. “魔鬼代言人”: 每当一个专家完成一项任务时,一个“验证员”智能体就会尝试破坏它。如果验证员发现缺陷,专家必须修复它。这个循环持续进行,直到工作无懈可击。

结论

论文表明,A3D 可以处理复杂的科学代码(例如用于分子动力学的 LAMMPS 或用于量子化学的 QMCPACK),并自动设计定制硬件加速器。在一次测试中,它甚至将原本为图形卡(GPU)编写的代码成功重新设计为定制硬件加速器,证明其能够处理最棘手的场景。

简而言之,A3D 是一个完全自主的 AI 智能体团队,它充当桥梁,将复杂的软件问题转化为定制的高速硬件解决方案,而无需人类工程师手把手指导。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →