SuperAPI AI 工具集

kernelgen-flagos

编程与开发

查看项目

更新于 10-15 21:42

技能介绍

用途:作为统一入口,把 GPU 算子的生成、优化、平台特化与反馈等子技能打包在一起,并根据当前仓库类型自动分派到对应工作流。

主要能力:生成子技能包括 kernelgen-generate.md(任意 Python/Triton 仓库)、kernelgen-generate-for-flaggems.md(FlagGems 专用)、kernelgen-generate-for-vllm.md(vLLM 专用);优化子技能包括 kernelgen-optimize.md(通用 Triton 内核迭代优化)、kernelgen-optimize-for-flaggems.md(FlagGems 三种模式:built-in/external/experimental)、kernelgen-optimize-for-vllm.md(含 CustomOp 注册);平台特化子技能为 kernelgen-specialize.md 与 kernelgen-specialize-for-flaggems.md(四种模式:vendor-ops/vendor-fused/override-builtin/experimental);另有 kernelgen-submit-feedback.md 用于反馈。

使用方式:先检测仓库类型,通过 Glob 查找 pyproject.toml、setup.py、setup.cfg 并用 Read 读取以确定项目名,同时检查特征目录(FlagGems 看 src/flag_gems/,vLLM 看 vllm/ 与 csrc/)。随后按需求读取相应子技能文件并完整遵循其说明;可用 Glob 定位子技能路径,如 **/skills/kernelgen-flagos/kernelgen-generate.md。用户明确指定子技能时应优先遵从。

限制与重要规则:必须先检测再分派,不得跳过;子技能文件须完整读取,不得混用。所有算子代码生成必须经 mcp__kernelgen-mcp__generate_kernel,优化用 mcp__kernelgen-mcp__optimize_kernel,平台特化用 mcp__kernelgen-mcp__specialize_kernel;不得自行生成 Triton 内核、PyTorch 封装或算子实现。若 MCP 未配置、不可达或重试失败,应停止并报告,不得回退为手写代码。其他平台、价格与效果资料暂未提供。