SetFixpipeNz2ndFlag
功能说明
在DataCopy(CO1->GM,参考功能说明)的流程中,会涉及到nz2nd的流程该接口用于设置FixpipeNz2nd相关的配置。
函数原型
__aicore__ inline void SetFixpipeNz2ndFlag(uint16_t ndNum, uint16_t srcNdStride, uint16_t dstNdStride);
参数说明
参数名称 |
输入/输出 |
含义 |
---|---|---|
ndNum |
输入 |
nd的数量,类型是uint16_t。 |
srcNdStride |
输入 |
是以分形大小为单位的源步长。 源相邻nz矩阵的偏移(头与头),取值范围:srcNdStride∈[1, 512],单位:fractal_size 1024B |
dstNdStride |
输入 |
目的相邻nd矩阵的偏移(头与头),取值范围:dstNdstride∈[0, 65535],单位:element |
支持的型号
Atlas A2训练系列产品/Atlas 800I A2推理产品
注意事项
dstNdStride不能为0。
返回值
无
调用示例
uint16_t ndNum = 2; uint16_t srcNdStride = 2; uint16_t dstNdStride = 1; SetFixpipeNz2ndFlag(ndNum, srcNdStride, dstNdStride);
父主题: 矩阵计算