aclnnGeGluV3Backward
支持的产品型号
- Atlas 推理系列产品。
- Atlas 训练系列产品。
- Atlas A2训练系列产品/Atlas 800I A2推理产品。
接口原型
每个算子分为，必须先调用“aclnnGeGluV3BackwardGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnGeGluV3Backward”接口执行计算。
aclnnStatus aclnnGeGluV3BackwardGetWorkspaceSize(const aclTensor *gradOutput, const aclTensor *self, const aclTensor *gelu, int64_t dim, int64_t approximate, bool activateLeft, aclTensor *gradInput, uint64_t *workspaceSize, aclOpExecutor **executor)aclnnStatus aclnnGeGluV3Backward(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
功能描述
aclnnGeGluV3BackwardGetWorkspaceSize
参数说明：
gradOutput(aclTensor*, 计算输入)：Device侧的aclTensor，数据类型支持FLOAT32、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品），数据类型与self一致, shape中除dim维外，其它维的大小跟self一样，dim维的大小是self的一半，支持，支持ND。
self(aclTensor*, 计算输入)：Device侧的aclTensor，数据类型支持FLOAT32、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品）, shape中除dim维外，其它维的大小跟gradOutput一样，dim维的大小是gradOutput的两倍，支持，支持ND。
gelu(aclTensor*, 计算输入)：Device侧的aclTensor，数据类型支持FLOAT32、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品），数据类型与self一致, shape需要与gradOutput一样，支持，支持ND。
dim(int64_t, 计算输入)：Host侧的整数，数据类型支持INT64，取值范围为[-self.dim(), self.dim()-1]。
approximate(int64_t, 计算输入)：Host侧的整数，数据类型支持INT64，取值范围是0('none')、1('tanh') 。
activateLeft(bool, 计算输入)：计算属性，host侧的布尔值，表示激活函数操作数据块的方向，默认值为false，表示对右边做activate。
gradInput(aclTensor*, 计算输出)：Device侧的aclTensor，数据类型支持FLOAT32、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品）, 数据类型与self一致, shape需要与self一样，支持，支持ND。
workspaceSize(uint64_t*, 出参)：返回需要在Device侧申请的workspace大小。
executor(aclOpExecutor**, 出参)：返回op执行器，包含了算子计算流程。
返回值：
[object Object]
aclnnGeGluV3Backward
参数说明：
workspace(void*, 入参)：在Device侧申请的workspace内存地址。
workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnGeGluV3BackwardGetWorkspaceSize获取。
executor(aclOpExecutor*, 入参)：op执行器，包含了算子计算流程。
stream(aclrtStream, 入参)：指定执行任务的AscendCL Stream流。
返回值：
约束与限制
无。