triton-operator-code-gen

Official

Generate 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 required

Components

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.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.