Commit 13941241 authored by Dimitris Vardoulakis's avatar Dimitris Vardoulakis Committed by TensorFlower Gardener
Browse files

[TF:XLA] Improve the accounting for subcomputations in the heap simulator. 

Subtract the size of the aliased buffers from the subcomputation estimate instead of from the current computation. This way, the memory estimate for the current computation is more accurate.

For the newly added test, the heap simulation calculates 48 bytes at head instead of the correct 64 bytes.

PiperOrigin-RevId: 215653047
parent aeb044c9
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