下载
EN
注册

SetFixPipeAddr

功能说明

DataCopy(CO1->GM)过程中进行随路量化后,通过调用该接口设置element-wise操作时LocalTensor的地址。

函数原型

1
2
template <typename T>
__aicore__ inline void SetFixPipeAddr(const LocalTensor<T> &eleWiseTensor, uint16_t c0ChStride)

参数说明

表1 参数说明

参数名称

输入/输出

含义

eleWiseTensor

输入

L1 Buffer上的源操作数。类型为LocalTensor。

支持的TPosition为A1/B1/C1。起始地址需要保证32字节对齐,仅支持half数据类型。

c0ChStride

输入

L1 Buffer上的C0 channel stride,单位是C0_SIZE(32B)。

eleWiseTensor沿N方向以C0为单位切分得到的数据块称为C0 channel,两块C0 channel的间隔称之为C0 channel stride。

支持的型号

Atlas 200I/500 A2 推理产品

注意事项

返回值

调用示例

完整示例可参考完整示例

1
2
3
4
__aicore__inline void SetEleSrcPara(const LocalTensor <half>& eleWiseTensor, uint16_t c0ChStride)
{
    AscendC::SetFixPipeAddr(eleWiseTensor, c0ChStride);
}