SetFixpipeNz2ndFlag
功能说明
DataCopy(CO1->GM、CO1->A1)过程中进行随路格式转换(NZ2ND)时,通过调用该接口设置NZ2ND相关配置。
函数原型
1
|
__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。
返回值
无
调用示例
完整示例可参考完整示例。
1 2 3 4 |
uint16_t ndNum = 2; uint16_t srcNdStride = 2; uint16_t dstNdStride = 1; AscendC::SetFixpipeNz2ndFlag(ndNum, srcNdStride, dstNdStride); |
父主题: 矩阵计算