SetFixPipeAddr
功能说明
DataCopy（CO1->GM）过程中进行随路量化后，通过调用该接口设置element-wise操作时LocalTensor的地址。
函数原型
1 2 | template <typename T> __aicore__ inline void SetFixPipeAddr(const LocalTensor<T> &eleWiseTensor, uint16_t c0ChStride) |
参数说明
参数名称 |
输入/输出 |
含义 |
|---|---|---|
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 200/500 A2推理产品
注意事项
无
返回值
无
调用示例
完整示例可参考完整示例。
1 2 3 4 | __aicore__inline void SetEleSrcPara(const LocalTensor <half>& eleWiseTensor, uint16_t c0ChStride) { AscendC::SetFixPipeAddr(eleWiseTensor, c0ChStride); } |
父主题： 矩阵计算
