下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

SetFixpipeNz2ndFlag

功能说明

在DataCopy(CO1->GM,参考功能说明)的流程中,会涉及到nz2nd的流程该接口用于设置FixpipeNz2nd相关的配置。

函数原型

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

参数说明

表1 通用参数说明

参数名称

输入/输出

含义

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);
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词