aclnnGeGluBackward
支持的产品型号
- Atlas 推理系列产品。
- Atlas 训练系列产品。
- Atlas A2训练系列产品/Atlas 800I A2推理产品。
接口原型
每个算子分为undefined，必须先调用“aclnnGeGluBackwardGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnGeGluBackward”接口执行计算。
aclnnStatus aclnnGeGluBackwardGetWorkspaceSize(const aclTensor *gradOutput, const aclTensor *self, const aclTensor *gelu, int64_t dim, int64_t approximate, aclTensor *gradInput, uint64_t *workspaceSize, aclOpExecutor **executor)aclnnStatus aclnnGeGluBackward(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
功能描述
算子功能：完成GeGlu的反向。
aclnnGeGluBackwardGetWorkspaceSize
参数说明：
gradOutput(aclTensor*,计算输入)：Device侧的aclTensor，数据类型支持FLOAT32、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品支持），数据类型与self一致, shape中除dim维外，其它维的大小跟self一样，dim维的大小是self的一半，支持undefined，undefined支持ND。
self(aclTensor*,计算输入)：Device侧的aclTensor，数据类型支持FLOAT32、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品支持）, shape中除dim维外，其它维的大小跟gradOutput一样，dim维的大小是gradOutput的两倍，支持undefined，undefined支持ND。
gelu(aclTensor*,计算输入)：Device侧的aclTensor，数据类型支持FLOAT32、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品支持），数据类型与self一致, shape需要与gradOutput一样，支持undefined，undefined支持ND。
dim(int64_t,计算输入)：Host侧的整数，数据类型支持INT64，当前取值支持[-len(输入shape),len(输入shape))。
approximate(int64_t,计算输入)：Host侧的整数，数据类型支持INT64，取值范围是0('none')、1('tanh') ，当前取值只支持1('tanh') 。
gradInput(aclTensor*,计算输出)：Device侧的aclTensor，数据类型支持FLOAT32、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品支持）, 数据类型与self一致, shape需要与self一样，支持undefined，undefined支持ND。
workspaceSize(uint64_t*,出参)：返回需要在Device侧申请的workspace大小。
executor(aclOpExecutor**,出参)：返回op执行器，包含了算子计算流程。
返回值：
aclnnStatus：返回状态码，具体参见undefined。
[object Object]
aclnnGeGluBackward
参数说明：
workspace(void*,入参)：在Device侧申请的workspace内存地址。
workspaceSize(uint64_t,入参)：在Device侧申请的workspace大小，由第一段接口aclnnGeGluBackwardGetWorkspaceSize获取。
executor(aclOpExecutor*,入参)：op执行器，包含了算子计算流程。
stream(aclrtStream,入参)：指定执行任务的 AscendCL Stream流。
返回值：
aclnnStatus：返回状态码，具体参见undefined。
约束与限制
无。
调用示例
示例代码如下，仅供参考，具体编译和执行过程请参考undefined。