昇腾社区首页
中文
注册

SetFixpipeNz2ndFlag(ISASI)

功能说明

DataCopy(CO1->GM、CO1->A1)过程中进行随路格式转换(NZ2ND)时,通过调用该接口设置NZ2ND相关配置。

函数原型

1
__aicore__ inline void SetFixpipeNz2ndFlag(uint16_t ndNum, uint16_t srcNdStride, uint16_t dstNdStride)

参数说明

表1 参数说明

参数名称

输入/输出

含义

ndNum

输入

nd的数量,类型是uint16_t,取值范围:ndNum∈[1, 65535]。

srcNdStride

输入

以分形大小为单位的源步长,源相邻nz矩阵的偏移(头与头)。

Atlas A2 训练系列产品/Atlas 800I A2 推理产品srcNdStride∈[1, 512],单位:fractal_size 1024B。

Atlas 200I/500 A2 推理产品srcNdStride∈[1, 512],单位:fractal_size 1024B。

dstNdStride

输入

目的相邻nd矩阵的偏移(头与头)。单位为元素。

Atlas A2 训练系列产品/Atlas 800I A2 推理产品,dstNdstride∈[1, 65535]。

Atlas 200I/500 A2 推理产品,dstNdstride∈[1, 65535]。

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品

Atlas 200I/500 A2 推理产品

注意事项

无。

返回值

调用示例

完整示例可参考完整示例

1
2
3
4
uint16_t ndNum = 2;
uint16_t srcNdStride = 2;
uint16_t dstNdStride = 1;
AscendC::SetFixpipeNz2ndFlag(ndNum, srcNdStride, dstNdStride);