使用说明

⚠️ 已废弃:CLI 模块(akg_cli)已停止维护,后续不再演进。

启动 AKG_CLI 后,您可以看到以下 AKG_CLI 启动界面,然后请通过以下方式使用:

AKG CLI启动界面

方式一:直接提问,与 CLI 直接交流,提出任务

步骤1:在输入框提出任务,点击回车,然后 CLI 就会开启思考帮您创建 task 代码,生成结束之后会询问用户接下来的需求

CLI task生成

步骤二:修改并确认task

如果用户对task的创建不满意,可以调整输入的shape大小,数据类型等 CLI task修改

步骤三:kernel生成与总结

我们提供了三种kernel的生成方式:

  • CoderOnly:标准的coder流程
  • Evolve:进化搜索,生成的kernel较coderonly流程的结果性能更佳
  • Adaptive Search:自适应搜索,选择策略的异步流水线搜索框架

温馨提示: 如果您对我们的 kernel 生成方式感兴趣,请查看《Evolve》、《Adaptive Search》。

  1. 请选择您需要的生成方式,并做出确认,即可开启kernel生成 CLI kernel生成

  1. kernel生成过程 在这里您能看到Agent执行的过程以及结果文件存储路径等信息 CLI kernel生成

  1. kernel生成结果 生成结束之后会展示选择的kernel生成的结果以及总结等信息,例如:我们选择的Adaptive Search的结果如下: CLI kernel生成

方式二:直接粘贴现有的 KernelBench 风格的代码作为baseline,AIKG会根据输入的task,校验并修正 baseline task 代码用于结果对比。

举例说明:

  1. 我们输入一个task代码: CLI kernel生成

  1. 修正后的结果如下,这个task将用于后面的kernel生成 CLI kernel生成
  • 说明:后续的kernel生成流程相信您已经掌握方式一中介绍的用法了,请参考使用。
  • 提示:AKG_CLI会持续更新与维护,如果您有宝贵的建议,请提issue