---
title: SetTensorA
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/ascendcopapi/atlasascendc_api_07_0631.html
sourcePath: /source/zh/canncommercial/900/API/ascendcopapi/atlasascendc_api_07_0631.html
indexId: eaa586365c6129530a3d0945e2ae09bd03bc1b944ebb1bab3c76079ee2880b2176
---
# SetTensorA

#### 产品支持情况

| 产品 | 是否支持 |
| --- | --- |
| Atlas 350 加速卡 | √ |
| Atlas A3 训练系列产品 / Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品 / Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | √ |
| Atlas 推理系列产品 AI Core | √ |
| Atlas 推理系列产品 Vector Core | x |
| Atlas 训练系列产品 | x |


#### 功能说明

设置矩阵乘的左矩阵A。


#### 函数原型

```
__aicore__ inline void SetTensorA(const GlobalTensor<SrcAT>& gm, bool isTransposeA = false)
```


```
__aicore__ inline void SetTensorA(const LocalTensor<SrcAT>& leftMatrix, bool isTransposeA = false)
```


```
__aicore__ inline void SetTensorA(SrcAT aScalar)
```


Atlas 推理系列产品 AI Core不支持SetTensorA(SrcAT aScalar)接口原型。

Atlas 200I/500 A2 推理产品 ，不支持SetTensorA(SrcAT aScalar)接口原型。


#### 参数说明


**表1 参数说明**

| 参数名 | 输入/输出 | 描述 |
| --- | --- | --- |
| gm | 输入 | A矩阵。类型为GlobalTensor。SrcAT参数表示A矩阵的数据类型。 Atlas 350 加速卡，支持的数据类型为：half/float/bfloat16\_t/int8\_t/fp8\_e4m3fn\_t/fp8\_e5m2\_t/hifloat8\_t Atlas A3 训练系列产品 / Atlas A3 推理系列产品 ，支持的数据类型为：half/float/bfloat16\_t/int8\_t/int4b\_t Atlas A2 训练系列产品 / Atlas A2 推理系列产品 ，支持的数据类型为：half/float/bfloat16\_t/int8\_t/int4b\_t Atlas 推理系列产品 AI Core，支持的数据类型为：half/float/int8\_t Atlas 200I/500 A2 推理产品 ，支持的数据类型为：half/float/bfloat16\_t/int8\_t |
| leftMatrix | 输入 | A矩阵。类型为LocalTensor，支持的TPosition为TSCM/VECOUT。SrcAT参数表示A矩阵的数据类型。 Atlas 350 加速卡，支持的数据类型为：half/float/bfloat16\_t/int8\_t/fp8\_e4m3fn\_t/fp8\_e5m2\_t/hifloat8\_t Atlas A3 训练系列产品 / Atlas A3 推理系列产品 ，支持的数据类型为：half/float/bfloat16\_t/int8\_t/int4b\_t Atlas A2 训练系列产品 / Atlas A2 推理系列产品 ，支持的数据类型为：half/float/bfloat16\_t/int8\_t/int4b\_t Atlas 推理系列产品 AI Core，支持的数据类型为：half/float/int8\_t Atlas 200I/500 A2 推理产品 ，支持的数据类型为：half/float/bfloat16\_t/int8\_t 若设置TSCM首地址，默认矩阵可全载，已经位于TSCM，Iterate接口无需再进行GM\->A1/B1搬运。 |
| aScalar | 输入 | A矩阵中设置的值。支持传入标量数据，标量数据会被扩展为一个形状为[1, K]的tensor参与矩阵乘计算，tensor的数值均为该标量值。例如，开发者可以通过将aScalar设置为1来实现矩阵B在K方向的reduce sum操作。SrcAT参数表示A矩阵的数据类型。 Atlas 350 加速卡，支持的数据类型为：half/float Atlas A3 训练系列产品 / Atlas A3 推理系列产品 ，支持的数据类型为：half/float Atlas A2 训练系列产品 / Atlas A2 推理系列产品 ，支持的数据类型为：half/float Atlas 推理系列产品 AI Core不支持该参数。 Atlas 200I/500 A2 推理产品 不支持该参数。 |
| isTransposeA | 输入 | A矩阵是否需要转置。 注意： 若A矩阵MatmulType的ISTRANS参数设置为true，该参数可以为true也可以为false，即运行时可以转置和非转置交替使用； 若A矩阵MatmulType的ISTRANS参数设置为false，该参数只能设置为false，若强行设置为true，精度会有异常； 对于非half、非bfloat16\_t输入类型的场景，为了确保Tiling侧与Kernel侧L1 Buffer空间计算大小保持一致及结果精度正确，该参数取值必须与Kernel侧定义A矩阵MatmulType的ISTRANS参数以及Tiling侧SetAType()接口的isTrans参数保持一致，即上述三个参数必须同时设置为true或同时设置为false。 Atlas 推理系列产品 AI Core，A矩阵为int8\_t数据类型时不支持转置，即不支持该参数设置为true。 Atlas A2 训练系列产品 / Atlas A2 推理系列产品 ，A矩阵为int4b\_t数据类型时不支持转置，即不支持该参数设置为true。 Atlas A3 训练系列产品 / Atlas A3 推理系列产品 ，A矩阵为int4b\_t数据类型时不支持转置，即不支持该参数设置为true。 |


#### 返回值说明

无


#### 约束说明

传入的TensorA地址空间大小需要保证不小于singleM * singleK。


#### 调用示例

```
REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling);
// 示例一：左矩阵在Global Memory
mm.SetTensorA(gm_a, isTransposeA);
mm.SetTensorB(gm_b);
if (tiling.isBias) {
    mm.SetBias(gmBias);
}
mm.IterateAll(gm_c);
mm.End();
// 示例二：左矩阵在Local Memory
mm.SetTensorA(local_a, isTransposeA);
// 示例三：设置标量数据
mm.SetTensorA(scalar_a, isTransposeA);
```