experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Erase cloned instructions on the fly when merging fusion nodes.

Browse Source 
This avoids the awkward situation where an RNG which is clearly eligible for fusion becomes ineligible mid-fusion because it suddenly has an extra (dead) user.

PiperOrigin-RevId: 173141716
This commit is contained in:
A. Unique TensorFlower 2017-10-23 11:12:26 -07:00 committed by TensorFlower Gardener
parent 1038927c09
commit 5e23e0e67a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/service/hlo_instruction.cc
View File

@ -716,10 +716,12 @@ void HloInstruction::MergeFusionInstructionIntoMultiOutput(
// Fuse the root instruction and generate multiple outputs. // Fuse the root instruction and generate multiple outputs.
FuseInstructionIntoMultiOutput(unfused_root); FuseInstructionIntoMultiOutput(unfused_root);
TF_CHECK_OK(unfused_root->parent()->RemoveInstruction(unfused_root));
// The rest instructions are of normal fusing. // The rest instructions are of normal fusing.
for (int64 i = 1; i < unfused_instructions.size(); i++) { for (int64 i = 1; i < unfused_instructions.size(); i++) {
auto instruction = unfused_instructions[i]; auto instruction = unfused_instructions[i];
FuseInstruction(instruction); FuseInstruction(instruction);
TF_CHECK_OK(instruction->parent()->RemoveInstruction(instruction));
} }
} }