OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Properly tiebreak ForceDelay

Browse Source 
PiperOrigin-RevId: 833065628
This commit is contained in:
A. Unique TensorFlower
2025-11-16 15:16:18 -08:00
committed by TensorFlower Gardener
parent 389512473e
commit 7507dadc02
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
third_party/xla/xla/service/latency_hiding_scheduler.cc vendored
View File

@@ -1278,10 +1278,13 @@ class ReadySetLt {
HloGraphNode* bn = b.node;
// Schedule according to ForceEarly.
CMP_PROPERTY(GetForceEarly(), "kForceEarly");
// Schedule according to highest ForceDelay first.
// Schedule according to ForceDelay, if exactly one of the two instructions
// has ForceDelay set.
CMP_EXPLICIT(!an->GetForceDelay(), !bn->GetForceDelay(), "kForceDelay");
// Schedule according to highest ForceDelay first, if both instructions
// have ForceDelay set.
CMP_EXPLICIT(-an->GetForceDelayPriority(), -bn->GetForceDelayPriority(),
"kForceDelayPriority");
CMP_EXPLICIT(!an->GetForceDelay(), !bn->GetForceDelay(), "kForceDelay");
// Use the preference value (comes from a heuristic) to choose between
// the two candidates. If two preferences are the same regular LHS logic
// will run as usual, we take advantage of this fact when initializing