第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/。
延伸阅读
- 待补:正式正文完成时补充对应官方文档、论文或工具链接。