rocm-kernels - 优化 AMD ROCm 管线中的 Triton 内核
指导在 ROCm 上为 AMD GPU 开发、集成、基准测试和调试优化的 Triton 内核。
标签
更新于: 2026-10-07能力
典型输入
典型输出
该技能可以做什么
- 编写优化的 Triton 内核
- 基准测试内核性能
- 集成 diffusers 管线
- 优化 AMD GPU 内核
- 调试 ROCm/HIP 问题
输入
- 内核类型
- Triton 或 PyTorch 源代码
- HuggingFace 管线或模型
- 基准测试参数
- AMD GPU 目标设备
输出
- 内核实现指导
- 基准测试数据
- JSON 基准测试结果
- 注入内核的管线
要求
- ROCm 环境
- AMD MI355X 或 R9700 GPU
- Triton
- PyTorch
- HuggingFace diffusers 或 transformers