trunk/2e84b817280c0b86e806cff7fb2959d9a80f05a9: [MPS] Fix large tensor skips for `grid_sample` and `bmm` tests (#199181)
- PyTorch: 1780 events in the last 90 days
- PyTorch: 1758th Release in the last 90 days
- Previous: earlier the same day · trunk/ce42d088103272f0a90317697749d2069c815959
What happened
Fixes #198698 Updates/adds largeTensorTest and largeMPSBufferTest decorators for test_large_bmm , test_grid_sample_large_index_2d , and test_grid_sample_large_index_3d . Memory usage was measured by adding the following decorator to the test functions: def _print_mps_memory_usage ( fn ): import functools @ functools . wraps ( fn ) def wrapper ( * args , ** kwargs ): torch . accelerator . synchronize () torch . accel…
Summary assembled by rule from the sources below