triton-operator-code-gen
OfficialGenerate Triton kernels from operator designs.
AuthorAscend
Version1.0.0
Installs0
System Documentation
What problem does it solve?
通过将 Ascend NPU 算子设计文档转化为 Triton kernel 代码,降低实现门槛并缩短从设计到可执行实现的周期,同时附带基础正确性测试以确保实现符合设计意图。
Core Features & Use Cases
- 将算子设计文档自动转译为可编译的 Triton kernel,并附带基本正确性测试。
- 支持输入/输出规格、分块策略和模板映射等工作流要点,帮助开发者快速落地实现。
- Use Case: 当需要实现新算子时,从设计输入生成为可运行的内核代码及烟雾测试,用于快速验证正确性。
Quick Start
Generate Triton kernel code from an Ascend NPU operator design and run the included smoke test with your design input.
Dependency Matrix
Required Modules
None requiredComponents
Standard package💻 Claude Code Installation
Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.
Please help me install this Skill: Name: triton-operator-code-gen Download link: https://github.com/Ascend/agent-skills/archive/main.zip#triton-operator-code-gen Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.