SuperAPI AI 工具集

perf-optimization

通用与其它

查看项目

更新于 10-16 07:24

技能介绍

用途:这是一份性能优化协调手册(playbook),用于把性能优化任务拆解、路由给合适的专家并验证结果。它本身不实现代码,只负责规划、协调与验证。

主要能力:定义五类专家——perf-torch-cuda-graph-specialist(图捕获与重放)、perf-profiling-specialist(性能验证与测量)、kernel-triton-specialist(从零编写 Triton 内核)、kernel-tileir-specialist(为 Blackwell GPU 优化已有 Triton 内核的 TileIR 后端)、kernel-cute-specialist(CuTe DSL 内核)。提供任务到专家的映射表、按用户意图的路由规则、迭代优化循环(优化→剖析→重复)、远程 SLURM 执行上下文复用规则,以及安全修改流程。

使用方式:用户指定优化时,解析请求、检查前置条件、制定计划、委派专家、前后测量、报告结果;自动驾驶模式下接收瓶颈分析(如 memory-bound 及建议列表),按影响/风险排序,逐项实施并在每步之间验证,出现回退则还原。修改前须先 backup_file,失败时 revert_file。TileIR 请求走两步流水线:先由 kernel-triton-specialist 生成基础 Triton 内核,再由 kernel-tileir-specialist 应用 TileIR 优化;已有内核可跳过第一步。

适用场景:CUDA Graph、内核融合、Triton/CuTe DSL 内核生成、TileIR 优化、分布式与并行策略等优化任务的协调。

限制:协调者不得自行编写内核、基准或脚本,也不得直接浏览或读取技能目录;TileIR 专家不从零写内核;CuTe DSL 与 Triton 任务不可混派;回退阈值约为性能下降超过 5% 或出现正确性问题。具体命令、安装方式与收费信息暂未提供。