triton-ascend-example-double-kernel

What problem does it solve?

该示例展示在 forward 中分两阶段调用两个 Triton Ascend kernel 的写法，解决需要中间结果缓冲与阶段性计算的算子实现难题。

Core Features & Use Cases

• 支持分阶段计算的算子结构，先执行 stage1 再执行 stage2，并确保阶段间数据正确传递。
• 提供中间缓冲区分配和两次 kernel 启动的标准模式，适用于需要分阶段计算的融合算子。
• 使用场景包括需要分步处理的融合算子和需要优化的管线化任务。

Quick Start

使用本示例在 forward 中按照顺序调用两个 kernel 来完成逐步计算。