GeluMul

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×
Atlas Kirin X90 处理器系列产品	√
Atlas Kirin 9030 处理器系列产品	√

功能说明

• 接口功能：

  将输入Tensor按照最后一个维度分为左右两个Tensor：x1和x2，对左边的x1进行GELU计算，将计算结果与x2相乘。

• 计算公式：

  给定输入张量input，最后一维的长度为2d，函数GeluMul进行以下计算：

  1. 将input分割为两部分：

     $$x_1=\text{input}[...,:d],x_2=\text{input}[...,d:]$$

  2. 对x1应用GELU激活函数，“tanh”模式公式如下：

     $$\text{GELU}(x)=0.5\cdot x\cdot \left(1+\tanh \left(\sqrt{\frac{2}{\pi }}\cdot \left(x+0.044715x^3\right)\right)\right)$$

     “none”对应的erf模式公式如下：

     $$\text{GELU}(x)=0.5\cdot x\left(1+\text{erf}\left(\frac{x}{\sqrt{2}}\right)\right)$$

     因此，计算：

     $$x_1=\text{GELU}(x_1)$$

  3. 最终输出是x1和x2的逐元素乘积：

     $$\text{out}=x_1\times x_2$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
input	输入	公式中的输入input。	BFLOAT16、FLOAT16、FLOAT	ND
approximateOptional	可选属性	GELU计算的模式，只支持“none”和“tanh”，分别对应GELU的erf模式和tanh模式，输入为空指针时为“none”。	STRING	-
out	输出	公式中的out。	BFLOAT16、FLOAT16、FLOAT	ND

• Atlas Kirin X90 处理器系列产品、Atlas Kirin 9030 处理器系列产品：不支持BFLOAT16。

约束说明

典型场景尾轴为16的倍数，当尾轴为非32B对齐时，建议走小算子拼接逻辑。

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_gelu_mul	通过aclnnGeluMul接口方式调用GeluMul算子。
图模式调用	-	通过算子IR构图方式调用GeluMul算子。