| #define TORCH_ASSERT_NO_OPERATORS |
| #include <ATen/cuda/CUDAGeneratorImpl.h> |
| #include <ATen/native/UnaryOps.h> |
| #include <ATen/native/cuda/DistributionTemplates.h> |
| |
| namespace at::native { |
| |
| void exponential_kernel(TensorIteratorBase& iter, double lambda, c10::optional<Generator> gen) { |
| auto generator = get_generator_or_default<CUDAGeneratorImpl>(gen, cuda::detail::getDefaultCUDAGenerator()); |
| at::native::templates::cuda::exponential_kernel(iter, lambda, generator); |
| } |
| |
| REGISTER_DISPATCH(exponential_stub, &exponential_kernel); |
| |
| } // namespace at::native |
