Commit 5529a326 authored by Justin Lebar's avatar Justin Lebar Committed by TensorFlower Gardener
Browse files

[XLA] Free the previous run's result before running the next iteration in replay_computation. 

Otherwise, we will try to allocate the current iteration's result while the
previous iter's result is still live.  If the result is large enough, this can
OOM us.

PiperOrigin-RevId: 218955957
parent 68ebb861
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment