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

#### 产品支持情况

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


#### 功能说明

MxMatmul场景(https://www.hiascend.comdocument/detail/zh/canncommercial/900/programug/Ascendcopdevg/atlas_ascendc_10_10029.html)，设置矩阵乘中右矩阵的量化系数矩阵scaleB。


#### 函数原型

```
__aicore__ inline void SetTensorScaleB(const GlobalTensor<ScaleT>& gm, bool isTransposeScaleB = true);
```


```
__aicore__ inline void SetTensorScaleB(const LocalTensor<ScaleT>& rightMatrix, bool isTransposeScaleB = true);
```


#### 参数说明


**表1 参数说明**

| 参数名 | 输入/输出 | 描述 |
| --- | --- | --- |
| gm | 输入 | 量化系数scaleB矩阵。类型为GlobalTensor。 Atlas 350 加速卡，支持的数据类型为：fp8\_e8m0\_t |
| rightMatrix | 输入 | 量化系数scaleB矩阵。类型为LocalTensor，支持的TPosition为TSCM/VECOUT。 Atlas 350 加速卡，支持的数据类型为：fp8\_e8m0\_t |
| isTransposeScaleB | 输入 | scaleB矩阵是否需要转置。 参数支持的取值如下： false：scaleB矩阵不转置。 true：默认值，scaleB矩阵转置。 注意： scaleB矩阵为NZ格式时，该参数只支持取值为true。 若scaleB矩阵的SCALE\_ISTRANS(https://www.hiascend.comdocument/detail/zh/canncommercial/900/programug/Ascendcopdevg/atlas\_ascendc\_10\_10029.html\#ZH\-CN\_TOPIC\_0000002562429869\_\_zh\-cn\_topic\_0000002270097206\_p19617394191)参数设置为true，除scaleB为NZ格式场景，该参数支持取值为true、false，即运行时scaleB矩阵可以转置和非转置交替使用。 若scaleB矩阵的SCALE\_ISTRANS(https://www.hiascend.comdocument/detail/zh/canncommercial/900/programug/Ascendcopdevg/atlas\_ascendc\_10\_10029.html\#ZH\-CN\_TOPIC\_0000002562429869\_\_zh\-cn\_topic\_0000002270097206\_p19617394191)参数设置为false，该参数只支持取值为false，若强行设置为true，精度会有异常； 对于有Bias输入的场景，为了确保Tiling侧与Kernel侧L1 Buffer空间计算大小保持一致及结果精度正确，该参数取值必须与Kernel侧定义B矩阵MatmulMxType的SCALE\_ISTRANS(https://www.hiascend.comdocument/detail/zh/canncommercial/900/programug/Ascendcopdevg/atlas\_ascendc\_10\_10029.html\#ZH\-CN\_TOPIC\_0000002562429869\_\_zh\-cn\_topic\_0000002270097206\_p19617394191)参数以及Tiling侧SetScaleBType()接口的isScaleTrans参数保持一致，即有Bias输入的场景，上述三个参数必须同时设置为true或同时设置为false。 |


#### 返回值说明

无


#### 约束说明

- 传入的scaleB地址空间大小必须不小于  singleCoreK
*  singleCoreN
/32。
- 当使能MixDualMaster（双主模式）场景时，即模板参数  enableMixDualMaster
设置为true，不支持使用该接口。


#### 调用示例

```
REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling);
mm.SetTensorA(gm_a);
mm.SetTensorB(gm_b);
mm.SetTensorScaleA(gm_scaleA);
mm.SetTensorScaleB(gm_scaleB);    // 设置右矩阵的量化系数矩阵scaleB
if (tiling.isBias) {
    mm.SetBias(gmBias);
}
mm.IterateAll(gm_c);
mm.End();
```