all_version: [v2.1, v2.2, v2.3, v2.4, v2.5, v2.6, v2.7, v2.8, v2.9, v2.10, v2.11, v2.12, v2.13]
official:
custom:
- func: npu_fast_gelu_custom(Tensor self) -> Tensor
op_api: all_version
gen_opapi:
out:
size: self
dtype: self
exec: aclnnFastGelu
- func: npu_fast_gelu_custom_backward(Tensor grad, Tensor self) -> Tensor
op_api: all_version
gen_opapi:
out:
size: grad
dtype: grad
exec: aclnnFastGeluBackward
autograd:
symint:
quant: