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

[XLA:CPU] Put the HLO name in IR values that hold the HLO's value.

Browse Source 
PiperOrigin-RevId: 171075449
This commit is contained in:
Justin Lebar 2017-10-04 15:44:14 -07:00 committed by TensorFlower Gardener
parent 89aaac4bc3
commit a02116882d
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/compiler/xla/service/cpu/ir_emitter.cc
View File

@ -2833,6 +2833,15 @@ Status IrEmitter::Preprocess(HloInstruction* hlo) {
}
Status IrEmitter::Postprocess(HloInstruction* hlo) {
// Set the name of the emitted llvm::Value to IrName(hlo). Outfeed and send
// the only ops that don't emit a value.
if (hlo->opcode() != HloOpcode::kOutfeed &&
hlo->opcode() != HloOpcode::kSend) {
auto it = emitted_value_.find(hlo);
CHECK(it != emitted_value_.end());
it->second->setName(AsStringRef(IrName(hlo)));
}
if (auto* prof_counter = GetProfileCounterFor(hlo)) {
profiling_state_.RecordCycleDelta(&ir_builder_, hlo, prof_counter);
}