SetBias

功能说明

设置矩阵乘的Bias。

函数原型

__aicore__ inline void SetBias(const GlobalTensor<BiasT>& biasGlobal)

__aicore__ inline void SetBias(const LocalTensor<BiasT>& inputBias)

参数说明

表1 参数说明
参数名	输入/输出	描述
biasGlobal	输入	Bias矩阵。类型为GlobalTensor。 Atlas 推理系列产品AI Core，支持的数据类型为：half/float/int32_t，其中仅在A、B的数据类型为int8_t时，Bias的数据类型可以设置为int32_t Atlas A2 训练系列产品/Atlas 800I A2 推理产品，支持的数据类型为：half/float/int32_t，其中仅在A、B的数据类型为int8_t时，Bias的数据类型可以设置为int32_t Atlas 200I/500 A2 推理产品，支持的数据类型为：half/float/int32_t，其中仅在A、B的数据类型为int8_t时，Bias的数据类型可以设置为int32_t A矩阵、B矩阵、Bias支持的数据类型组合可参考Matmul输入输出数据类型的组合说明。
inputBias	输入	Bias矩阵。类型为LocalTensor，支持的TPosition为TSCM/VECOUT。 Atlas 推理系列产品AI Core，支持的数据类型为：half/float/int32_t Atlas A2 训练系列产品/Atlas 800I A2 推理产品，支持的数据类型为：half/float/int32_t，其中仅在A、B的数据类型为int8_t时，Bias的数据类型可以设置为int32_t Atlas 200I/500 A2 推理产品，支持的数据类型为：half/float/int32_t，其中仅在A、B的数据类型为int8_t时，Bias的数据类型可以设置为int32_t A矩阵、B矩阵、Bias支持的数据类型组合可参考Matmul输入输出数据类型的组合说明。

返回值

无

支持的型号

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

Atlas 推理系列产品AI Core

Atlas 200I/500 A2 推理产品

约束说明

传入的Bias地址空间大小需要保证不小于singleN。

调用示例

REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling);
mm.SetTensorA(gm_a);
mm.SetTensorB(gm_b);
mm.SetBias(gm_bias);    // 设置Bias
mm.IterateAll(gm_c);

父主题： Matmul