brkirch 35b1775b32 Use other MPS optimization for large q.shape[0] * q.shape[1]
Check if q.shape[0] * q.shape[1] is 2**18 or larger and use the lower memory usage MPS optimization if it is. This should prevent most crashes that were occurring at certain resolutions (e.g. 1024x1024, 2048x512, 512x2048).

Also included is a change to check slice_size and prevent it from being divisible by 4096 which also results in a crash. Otherwise a crash can occur at 1024x512 or 512x1024 resolution.
2022-12-20 21:30:00 -05:00
..
2022-12-09 23:03:06 +09:00
2022-09-05 23:08:06 +03:00
2022-12-03 18:06:33 +03:00
2022-11-17 00:08:21 -05:00
2022-12-03 18:06:33 +03:00
2022-09-07 12:32:28 +03:00
2022-12-03 18:45:51 +03:00
2022-12-03 18:06:33 +03:00
2022-11-11 21:14:10 +07:00
2022-11-26 16:10:46 +03:00
2022-10-19 11:52:12 +03:00