开发者
下载
[object Object]

[object Object][object Object]
  • [object Object]Ascend 950PR/Ascend 950DT[object Object]:支持
[object Object][object Object]
  • [object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:不支持
[object Object][object Object]
  • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]:不支持
[object Object][object Object]
  • [object Object]Atlas 200I/500 A2 推理产品[object Object]:不支持
[object Object][object Object]
  • [object Object]Atlas 推理系列产品[object Object]:不支持
[object Object][object Object]
  • [object Object]Atlas 训练系列产品[object Object]:不支持
[object Object][object Object]
  • 接口功能:RmsNorm算子是大模型常用的标准化操作,相比LayerNorm算子,其去掉了减去均值的部分。RmsNormQuantV3算子将RmsNorm算子以及RmsNorm后的Quantize算子融合起来,减少搬入搬出操作。同时在RmsNormQuantV2算子的基础上新增了Rstd的输出。
  • 计算公式:quant_ini=xiRms(x)gammai+betai, where Rms(x)=1ni=1nxi2+epsilonquant\_in_i=\frac{x_i}{\operatorname{Rms}(\mathbf{x})} gamma_i + beta_i, \quad \text { where } \operatorname{Rms}(\mathbf{x})=\sqrt{\frac{1}{n} \sum_{i=1}^n x_i^2+epsilon}
    • divMode为True时:

      y=round((quant_in/scale)+offset)y=round((quant\_in/scale)+offset)
    • divMode为False时:

      y=round((quant_inscale)+offset)y=round((quant\_in*scale)+offset)
[object Object]

每个算子分为,必须先调用“aclnnRmsNormQuantV3GetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnRmsNormQuantV3”接口执行计算。

[object Object]
[object Object]
[object Object]
  • 参数说明

    [object Object]
  • 返回值

    aclnnStatus:返回状态码,具体参见

    第一段接口完成入参校验,出现以下场景时报错:

    [object Object][object Object]
[object Object]
  • 参数说明

    [object Object]
  • 返回值

    aclnnStatus:返回状态码。(具体参见

[object Object][object Object]
  • [object Object]Ascend 950PR/Ascend 950DT[object Object]:当[object Object]的数据类型为INT4时,[object Object][object Object]以及[object Object]的最后一维必须为偶数。
[object Object][object Object]
  • [object Object]Ascend 950PR/Ascend 950DT[object Object]:当[object Object]的数据类型为INT32时,[object Object]的最后一维必须是[object Object]最后一维的1/8。
[object Object]
  • 各产品型号支持数据类型说明:

    [object Object]
    • [object Object]Ascend 950PR/Ascend 950DT[object Object]:

      [object Object]undefined
    [object Object]
  • 确定性计算:

    • aclnnRmsNormQuantV3默认确定性实现。
[object Object]

示例代码如下,仅供参考,具体编译和执行过程请参考

[object Object]