FlagOS_KernelGen
服务介绍
KernelGen MCP Server
简介 | Introduction
KernelGen MCP Server 是由智源研究院打造的一款面向 Triton 开发者的代码自动生成与优化工具。它融合了智源研究院以及多家合作厂商在 Triton 领域的专家经验，致力于提升算子开发效率与性能表现。
KernelGen MCP Server is a powerful tool designed for Triton developers to automatically generate, optimize, and specialize operator code. It integrates expert knowledge from the Beijing Academy of Artificial Intelligence (BAAI) and industry partners to deliver high-quality, high-performance solutions.
该工具支持多种国内外芯片平台，具备高精度代码生成能力与优秀的执行性能，适用于 Triton 开发者及相关爱好者进行本地算子开发。
It supports a wide range of hardware platforms and delivers high-precision code generation with fast execution performance, making it ideal for Triton developers and enthusiasts working on custom operator development.
核心能力 | Key Capabilities
算子开发自动化 | Automated Operator Development
-
算子代码自动生成
支持通过自然语言描述算子需求，自动生成 Triton 代码
Generate Triton operator code directly from natural language descriptions -
算子代码错误自动修复
根据本地编译或测试日志自动修复代码错误
Automatically fix code based on compilation or test logs -
算子代码性能自动优化
根据日志分析优化代码，提高加速比
Optimize code for better acceleration performance based on runtime logs
多硬件适配 | Multi-Hardware Adaptation
- 国产芯片自动适配
自动适配华为昇腾、天数、海光、摩尔线程、沐曦股份等国产芯片
Automatically adapts code for various domestic chips (e.g., Ascend, Hygon, Moore Threads, etc.)
测试能力自动化 | Automated Testing
-
测试样例自动生成
根据算子描述生成多类型测试用例
Generate diverse test cases from operator descriptions -
精度测试代码自动生成
自动生成精度测试代码并输出测试结果
Generate accuracy test scripts and reports -
性能测试代码自动生成
自动生成加速比测试代码，对比 CUDA 算子性能
Generate performance benchmarks comparing against CUDA implementations
一键开发流程 | One-Click Workflow
以上所有功能均可通过 MCP Server 一键完成，实现 Triton 算子开发全流程自动化。
All the above capabilities can be executed in a single workflow via the MCP Server, enabling fully automated Triton operator development.
生态集成 | Ecosystem Integration
KernelGen MCP Server 支持与多种开发工具和 AI Agent 集成使用：
- VS Code（结合 GitHub Copilot）
- Claude Code
- OpenClaw
同时支持结合第一方与第三方 Skills 扩展能力：
更多信息请访问官网：
https://kernelgen.flagos.io/
使用前准备 | Prerequisites
在使用 KernelGen MCP Server 前，需获取 Bearer Token：
Before connecting your AI agent to the KernelGen MCP Server, you must obtain a Bearer Token.
获取步骤 | Steps to Retrieve Token
-
注册并登录 KernelGen Web Platform
Sign up and log in to the KernelGen Web Platform -
点击左上角导航栏中的 MCP Service
Click "MCP Service" in the top-left navigation menu -
在 Config 区域点击 Copy 获取 Bearer Token
Click "Copy" in the Config section to copy your Bearer Token
使用说明 | Usage
KernelGen MCP Server 可单独使用，也可结合 Skills 使用以增强功能：
KernelGen MCP Server can be used independently or combined with skills for enhanced capabilities.
- 单独使用：完成算子生成、优化、测试等任务
- 搭配 Skills：实现更复杂、更定制化的开发流程
总结 | Summary
KernelGen MCP Server 提供了一站式 Triton 算子开发解决方案，从代码生成到优化再到测试，全面提升开发效率与性能表现。
KernelGen MCP Server delivers an end-to-end solution for Triton operator development, significantly improving productivity and performance through automation and intelligent optimization.
