1 #define TORCH_ASSERT_NO_OPERATORS 2 #include <ATen/cuda/CUDAGeneratorImpl.h> 3 #include <ATen/native/UnaryOps.h> 4 #include <ATen/native/cuda/DistributionTemplates.h> 5 6 namespace at::native { 7 exponential_kernel(TensorIteratorBase & iter,double lambda,std::optional<Generator> gen)8void exponential_kernel(TensorIteratorBase& iter, double lambda, std::optional<Generator> gen) { 9 auto generator = get_generator_or_default<CUDAGeneratorImpl>(gen, cuda::detail::getDefaultCUDAGenerator()); 10 at::native::templates::cuda::exponential_kernel(iter, lambda, generator); 11 } 12 13 REGISTER_DISPATCH(exponential_stub, &exponential_kernel); 14 15 } // namespace at::native 16