Commit a6bb25c0 authored by Mark Heffernan's avatar Mark Heffernan Committed by TensorFlower Gardener
Browse files

Make scheduling and rematerialization HLO passes. 

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
parent 4136bd49
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