From 96eb2ab8ce3e0ec7dfc0de2dca19a0e6fa91c94c Mon Sep 17 00:00:00 2001 From: Justin Lebar Date: Fri, 3 Aug 2018 13:45:45 -0700 Subject: [XLA:GPU] Add VLOG to ForThunk. PiperOrigin-RevId: 207317857 --- tensorflow/compiler/xla/service/gpu/for_thunk.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow/compiler/xla/service/gpu/for_thunk.cc b/tensorflow/compiler/xla/service/gpu/for_thunk.cc index b3a3c5dcb4..2fd2206324 100644 --- a/tensorflow/compiler/xla/service/gpu/for_thunk.cc +++ b/tensorflow/compiler/xla/service/gpu/for_thunk.cc @@ -43,6 +43,8 @@ Status ForThunk::Initialize(const GpuExecutable& executable, Status ForThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, se::Stream* stream, HloExecutionProfiler* profiler) { + VLOG(2) << "Executing ForThunk with " << loop_limit_ << " iters for " + << (hlo_instruction() ? hlo_instruction()->ToString() : ""); auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); for (int64 i = 0; i < loop_limit_; ++i) { profiler->StartHloComputation(); -- cgit v1.2.3