Copy
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
功能说明
执行数据复制操作，通过CopyAtom选择不同存储空间之间的数据搬运。Copy API是Tensor API中的统一数据搬运接口，支持在不同存储空间之间进行数据传输。
表 1 支持的数据搬运通路
| 操作类型 | 说明 |
|---|---|
| CopyGM2L1 | Global Memory到L1 Buffer |
| CopyL12L0A | L1 Buffer到L0A Buffer |
| CopyL12L0B | L1 Buffer到L0B Buffer |
| CopyL12BT | L1 Buffer到BiasTable Buffer |
| CopyL12FB | L1 Buffer到Fixpipe Buffer |
| CopyL0C2GM | L0C Buffer搬出到Global Memory |
| CopyL0C2UB | L0C Buffer搬出到Unified Buffer |
函数原型
template <typename Tp, const Tp& traits, typename T, typename... Params>
__aicore__ inline void Copy(const CopyAtom<T>& atomCopy, const Params& ...params)
template <typename T, typename... Params>
__aicore__ inline void Copy(const CopyAtom<T>& atomCopy, const Params& ...params)
参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| atomCopy | 输入 | CopyAtom对象，封装了具体的复制操作和traits配置。 |
| params | 输入 | 复制操作的参数，可变参数。通常包括： - 目的张量（dst tensor） - 源张量（src tensor） - 可选：坐标偏移（Coord） |
返回值说明
无
约束说明
- atomCopy必须是有效的CopyAtom对象；
- params的数量和类型必须与复制操作的要求匹配；
- 源操作数和目的操作数的内存空间必须支持对应的复制操作。
调用示例
using namespace AscendC::Te;
// 以CopyL0C2GM为例
// 调用方式1
Copy(CopyAtom<CopyTraits<CopyL0C2GM, FixpipeTraitDefault>>{}, globalC, l0CTensor);
// 调用方式2
CopyAtom<CopyTraits<CopyL0C2GM, FixpipeTraitDefault>>{}.Call(globalC, l0CTensor);
// 调用方式3
auto atomCopyL0C2GM = MakeCopy(CopyL0C2GM{}, FixpipeTraitDefault{});
atomCopyL0C2GM.Call(globalC, l0CTensor);