⚠️ Alpha内测版本警告:此为早期内部构建版本,尚不完整且可能存在错误,欢迎大家提Issue反馈问题或建议。
Skip to content

第14章 算子优化 Agent 设计

本章导读

本章把前面封装的工具组装成一个完整的算子优化 Agent。读完后,你应该能理解 Agent 如何从一道算子题目出发,自动生成 kernel、跑 benchmark、根据结果反思改写。

14.1 Agent 的整体架构

画出 读题 → 生成 → 编译 → 跑分 → 反思 的循环架构图。

14.2 读题与问题理解

让 Agent 解析题目规格(输入形状、数据类型、期望性能)。

14.3 生成初始 kernel

让 Agent 根据题目生成第一版 naive kernel 作为 baseline。

14.4 跑分与性能反馈

调用 benchmark 工具拿到延迟/带宽,转成 Agent 能理解的反馈。

14.5 反思与改写

让 Agent 根据 profiling 信号(访存瓶颈?计算瓶颈?)决定下一步优化方向。

14.6 迭代终止条件

说明什么时候停(达到目标性能、迭代轮次上限、连续无提升)。

本章小结

  • 本章目前是 Alpha 阶段的大纲骨架,正式正文会在对应实验跑通后补齐。
  • 涉及命令、输出或性能数字的内容,后续必须在 Radeon RX 9070 XT + ROCm 7.13 / 原生 Ubuntu 24.04 上实测。
  • 与本章相关的代码、日志和实验底稿会放在 code/part3-agent/chapter14/

延伸阅读

  • 待补:正式正文完成时补充对应官方文档、论文或工具链接。