[XLA] Account for private stack allocations in HloRematerialization
[XLA] Account for private stack allocations in HloRematerialization

Buffer assignment allocates a single stack for all asynchronous computations, which persists for the entire duration of the program. We need to account for this by adjusting the memory limit.
