设置矩阵乘的左矩阵A。
__aicore__ inline void SetTensorA(const GlobalTensor<SrcT>& gm, bool isTransposeA = false);
__aicore__ inline void SetTensorA(const LocalTensor<SrcT>& leftMatrix, bool isTransposeA = false);
参数名 |
输入/输出 |
描述 |
---|---|---|
gm |
输入 |
A矩阵在Global Memory上的首地址。 Atlas 训练系列产品,支持的数据类型为:float Atlas推理系列产品AI Core,支持的数据类型为:float Atlas A2训练系列产品,支持的数据类型为:half/float |
leftMatrix |
输入 |
A矩阵在TSCM上的首地址或者在VECCALC上的首地址。 Atlas 训练系列产品,支持的数据类型为:float Atlas推理系列产品AI Core,支持的数据类型为:float Atlas A2训练系列产品,支持的数据类型为:half/float
|
isTransposeA |
输入 |
A矩阵是否需要转置。 |
无
Atlas A2训练系列产品
传入的TensorA地址空间大小需要保证不小于singleM * singleK。