Scatter API样例介绍

概述

本样例集介绍了Scatter API不同特性的典型用法,给出了对应的端到端实现。

算子开发样例

目录名称 功能描述
scatter 本样例介绍兼容Scatter算子实现及核函数直调方法,不支持Scatter能力的AI处理器,可以使用标量搬出的方式进行算子层面的仿真处理和修改算法的方式改变当前使用Scatter指令的算法。本样例是通过标量搬出的方式,使用<<<>>>内核调用符来完成算子核函数在NPU侧运行验证的基础流程,给出了对应的端到端实现
scatter_950 本样例基于Scatter指令实现数据离散,可用于根据一个输入张量、一个目的地址偏移张量和偏移地址,将输入张量分散到结果张量中。