#if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION) #pragma once #include #include namespace at { struct TensorIteratorBase; class TensorBase; } namespace at::native { void launch_glu_backward_kernel(const TensorIteratorBase& iter, int64_t gI_stride, int64_t I_stride); void launch_log_sigmoid_forward_kernel(TensorIteratorBase& iter); void GeluCUDAKernelImpl(TensorIteratorBase& it, GeluType approximate); void GeluBackwardCUDAKernelImpl(TensorIteratorBase& it, GeluType approximate); } // namespace at::native #else #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined." #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)