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

don't "print xla expression" in trunk annotation.

Browse Source 
PiperOrigin-RevId: 304090402
Change-Id: Iff60d4b92ce38d8e1d6204b5013f5b8d9fcf459b
This commit is contained in:
A. Unique TensorFlower 2020-03-31 18:33:38 -07:00 committed by Qiumin Xu
parent 53778b7e92
commit 85f156915b
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/xla/service/gpu/gpu_executable.cc
View File

@ -96,10 +96,8 @@ void GpuExecutable::ComputeThunkAnnotations() {
const HloInstruction* hlo = thunk->hlo_instruction(); const HloInstruction* hlo = thunk->hlo_instruction();
CHECK(hlo); CHECK(hlo);
thunk_annotations_[thunk] = thunk_annotations_[thunk] =
absl::StrFormat("%s:#hlo_op=%s,hlo_module=%s#", absl::StrFormat("Thunk#hlo_op=%s,hlo_module=%s#", hlo->name(),
hlo->ToStringWithCanonicalNameMap( hlo->GetModule()->name());
HloPrintOptions::Canonical(), &canonical_name_map),
hlo->name(), hlo->GetModule()->name());
} }
} }