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

Do not record queue length in metrics on task dequeue.

Browse Source 
It is more meaningful for the metrics to contain the queue length on enqueue tasks only.

PiperOrigin-RevId: 335682185
Change-Id: I40d2ed7555eb23bf8f62d7930186d6d0c49d275f
This commit is contained in:
Jing Dong 2020-10-06 11:26:05 -07:00 committed by TensorFlower Gardener
parent c5d4acd09a
commit 31029280a2
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/core/common_runtime/executor.cc
View File

@ -440,8 +440,7 @@ void ExecutorState<PropagatorStateType>::RunTask(Closure&& c) {
// mutable is needed because std::forward<Closure> in the lambda body may move // mutable is needed because std::forward<Closure> in the lambda body may move
// the Closure `c`. // the Closure `c`.
runner_([c = std::forward<Closure>(c)]() mutable { runner_([c = std::forward<Closure>(c)]() mutable {
auto qlen = queue_length.fetch_sub(1, std::memory_order_relaxed); queue_length.fetch_sub(1, std::memory_order_relaxed);
metrics::UpdateGraphPendingQueueLength(qlen - 1);
std::forward<Closure>(c)(); std::forward<Closure>(c)();
}); });
} }