trunk/2e84b817280c0b86e806cff7fb2959d9a80f05a9: [MPS] Fix large tensor skips for `grid_sample` and `bmm` tests (#199181)

发生了什么

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…

摘要按规则整理自下方来源原文

来源