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
cauchy_kernel(TensorIteratorBase & iter,double median,double sigma,std::optional<Generator> gen)8 void cauchy_kernel(TensorIteratorBase& iter, double median, double sigma, std::optional<Generator> gen) {
9 auto generator = get_generator_or_default<CUDAGeneratorImpl>(gen, cuda::detail::getDefaultCUDAGenerator());
10 at::native::templates::cuda::cauchy_kernel(iter, median, sigma, generator);
11 }
12
13 REGISTER_DISPATCH(cauchy_stub, &cauchy_kernel);
14
15 } // namespace at::native
16