diff --git a/aten/src/ATen/native/mps/operations/Distributions.mm b/aten/src/ATen/native/mps/operations/Distributions.mm index 7befd42b58ea..d19c2bd9994f 100644 --- a/aten/src/ATen/native/mps/operations/Distributions.mm +++ b/aten/src/ATen/native/mps/operations/Distributions.mm @@ -412,8 +412,8 @@ Tensor& randperm_out_mps(int64_t n, c10::optional generator, Tensor& }; return mps::random_mps_impl(result, - 0.0, - 1.0, + std::numeric_limits::min(), + std::numeric_limits::max(), c10::nullopt, c10::nullopt, MPSGraphRandomDistributionUniform,