diff --git a/src/ATen/native/xpu/sycl/DistributionTemplates.h b/src/ATen/native/xpu/sycl/DistributionTemplates.h index f70b47a305..34c7a9d912 100644 --- a/src/ATen/native/xpu/sycl/DistributionTemplates.h +++ b/src/ATen/native/xpu/sycl/DistributionTemplates.h @@ -472,11 +472,7 @@ void random_from_to_kernel( iter.dtype(), "random_from_to_kernel_xpu", AT_WRAP([&] { - if ((std::is_same::value || - std::is_same::value || - std::is_same::value || - std::is_same::value) && - range >= 1ULL << 32) { + if (range >= 1ULL << 28) { distribution_nullary_kernel< scalar_t, uint64_t,