pytorch/torch/testing
Nikita Shulga 1abfa5f70b [EZ][MPS] Improve distribution error checking (#166425)
Essentially not allow ops on self-overlapping outputs, by adding
`at::assert_no_internal_overlap(self);` check that already used in CPU
and CUDA builds, see
895795f07c/aten/src/ATen/native/DistributionTemplates.h (L366)

This fixes `test_error_inputs_bernoulli_mps`

Should be landed ahead of https://github.com/pytorch/pytorch/pull/165267
Pull Request resolved: https://github.com/pytorch/pytorch/pull/166425
Approved by: https://github.com/Skylion007, https://github.com/seemethere
2025-10-28 18:42:12 +00:00
..
_internal [EZ][MPS] Improve distribution error checking (#166425) 2025-10-28 18:42:12 +00:00
__init__.py Add pyrefly suppressions 2/n (#164513) 2025-10-03 02:46:13 +00:00
_comparison.py [1/N] Add strict parameter to Python zip calls (#165531) 2025-10-18 05:26:33 +00:00
_creation.py [BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552) 2025-08-07 00:09:56 +00:00
_utils.py Fix code descriptions in the test package. (#148145) 2025-03-04 19:14:41 +00:00