You can use MemoryFailPoint to check before a large distribution. This is not exactly what you are asking for, but if your failure is at a certain known point of high allocation, it at least allows you to clearly handle the distribution failure at the point at which it occurs.
source share