- [object Object]Ascend 950PR/Ascend 950DT[object Object]:支持
- [object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:不支持
- [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]:不支持
- [object Object]Atlas 200I/500 A2 推理产品[object Object]:不支持
- [object Object]Atlas 推理系列产品[object Object]:不支持
- [object Object]Atlas 训练系列产品[object Object]:不支持
- 接口功能:RmsNorm算子是大模型常用的标准化操作,相比LayerNorm算子,其去掉了减去均值的部分。RmsNormQuantV3算子将RmsNorm算子以及RmsNorm后的Quantize算子融合起来,减少搬入搬出操作。同时在RmsNormQuantV2算子的基础上新增了Rstd的输出。
- 计算公式:
divMode为True时:
divMode为False时:
每个算子分为,必须先调用“aclnnRmsNormQuantV3GetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnRmsNormQuantV3”接口执行计算。
[object Object]
[object Object]
参数说明
[object Object]返回值
第一段接口完成入参校验,出现以下场景时报错:
[object Object][object Object]
- [object Object]Ascend 950PR/Ascend 950DT[object Object]:当
[object Object]的数据类型为INT4时,[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]Ascend 950PR/Ascend 950DT[object Object]:
[object Object]undefined
确定性计算:
- aclnnRmsNormQuantV3默认确定性实现。
[object Object]