fla-triton-to-gluon - 将 Triton 内核移植到 Gluon
将 fla/ops/** 中的 Triton 内核移植到 Gluon,并显式控制张量布局、共享内存、异步数据移动、MMA 和调度。
标签
更新于: 2026-10-01能力
典型输入
典型输出
该技能可以做什么
- 评估移植可行性
- 将 Triton API 转换为 Gluon
- 添加显式张量布局
- 实现异步内存流水线
- 集成 MMA 指令
- 保持前向和反向一致性
- 分析并优化内核
输入
- 现有 Triton 内核
- 冻结的 pytest 契约
- 已安装的 Triton 版本
- 目标 NVIDIA GPU 架构
- 内核性能分析数据
输出
- Gluon 内核源代码
- 一致性测试结果
- 内核性能测量结果
要求
- 支持 Gluon 的 Triton 环境
- NVIDIA GPU 环境
- 支持所选特性的硬件
- 可访问内核测试套件
