| Commit message (Collapse) | Author | Age |
|
|
|
|
|
|
|
| |
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
|
|
|
|
| |
PiperOrigin-RevId: 215272497
|
|
Now that HloSchedule is a field on the HLO module, scheduling can be done as an HLO pass. Similarly, rematerialization which requires a schedule can also be a pass which just gets the schedule from the module.
Also as a clean up, hoist calls to CopyInsertion out of rematerialization.
PiperOrigin-RevId: 212119795
|