triton-ascend-fused-operator-optimization

Generate deterministic fusion plans for Ascend NPU operators.

258|48|Updated Jun 22, 2020
One-click install
npx skills add https://github.com/mindspore-ai/akg --skill triton-ascend-fused-operator-optimization-mindspore-ai
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: triton-ascend-fused-operator-optimization
Source: https://github.com/mindspore-ai/akg/tree/main/akg_agents/python/akg_agents/op/resources/skills/triton-ascend/evolved-improvement/triton-ascend-fused-operator-optimization
Command: npx skills add https://github.com/mindspore-ai/akg --skill triton-ascend-fused-operator-optimization-mindspore-ai

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Ascend NPU 上融合算子的深度优化方法论,覆盖性能天花板分析框架、多 Pass 合并策略、数据访问模式重构、Normalization 两阶段决策、NPU 原生算子评估方法论。适用于 elementwise 融合、归一化融合、softmax+topk 融合、matmul+activation 融合等场景。

Core Features & Use Cases

  • Performance ceiling analysis and plan for multi-pass fusion
  • Data access pattern restructuring to improve memory bandwidth
  • Two-stage normalization decision workflow (statistic + normalize)
  • Native operator evaluation methodology for Ascend NPU
  • Use Case: optimizing a fused elementwise + activation kernel on Ascend backends.

Quick Start

Provide an optimized fused-operator plan for Ascend NPU based on your operator graph and input shapes.

Frequently Asked Questions about triton-ascend-fused-operator-optimization

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I optimize fused operators for Ascend NPU?

Optimizing fused operators for Ascend NPU requires applying performance-ceiling analysis and multi-pass fusion strategies to generate a deterministic fusion plan based on operator graphs, input shapes, and data types.

What is the two-stage normalization decision workflow for Ascend NPU?

The two-stage normalization decision workflow on Ascend NPU separates the normalization process into distinct statistic and normalize passes to optimize data access patterns and improve memory bandwidth utilization.

How do I improve memory bandwidth for elementwise and matmul activation fusion?

Improving memory bandwidth for elementwise and matmul activation fusion involves restructuring data access patterns within the operator graph to minimize overhead across varying data layouts on Ascend backends.

Does Ascend NPU fused operator optimization support softmax and topk fusion scenarios?

Fused operator optimization explicitly supports softmax and topk fusion scenarios, alongside elementwise, normalization, and matmul activation fusion, by evaluating native Ascend NPU operators and applying multi-pass strategies.

What inputs are needed to generate an Ascend NPU fusion plan?

Generating a deterministic fusion plan requires providing the operator graph, input shapes, data types, and accuracy constraints to evaluate the performance ceiling and apply appropriate multi-pass merging strategies.

When should I use multi-pass fusion strategies for Ascend NPU kernels?

Multi-pass fusion strategies should be used when optimizing complex kernels like elementwise and normalization operations on Ascend NPU, where restructuring data access patterns can break through the performance ceiling.