1 #pragma once 2 #include <ATen/native/Activation.h> 3 #include <cstdint> 4 5 namespace at { 6 struct TensorIteratorBase; 7 class TensorBase; 8 } 9 10 namespace at { namespace native { 11 12 void launch_glu_backward_kernel(const TensorIteratorBase& iter, 13 int64_t gI_stride, int64_t I_stride); 14 15 void launch_log_sigmoid_forward_kernel(TensorIteratorBase& iter); 16 17 void GeluCUDAKernelImpl(TensorIteratorBase& it, GeluType approximate); 18 void GeluBackwardCUDAKernelImpl(TensorIteratorBase& it, GeluType approximate); 19 20 }} // namespace at::native 21