diff options
Diffstat (limited to 'src/zencompute/runners/functionrunner.cpp')
| -rw-r--r-- | src/zencompute/runners/functionrunner.cpp | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/src/zencompute/runners/functionrunner.cpp b/src/zencompute/runners/functionrunner.cpp index f3d5acca5..ab22c6363 100644 --- a/src/zencompute/runners/functionrunner.cpp +++ b/src/zencompute/runners/functionrunner.cpp @@ -6,7 +6,9 @@ # include <zencore/compactbinary.h> # include <zencore/filesystem.h> +# include <zencore/fmtutils.h> # include <zencore/logging.h> +# include <zencore/string.h> # include <zencore/timer.h> # include <zencore/trace.h> # include <zencore/workthreadpool.h> @@ -206,16 +208,24 @@ BaseRunnerGroup::SubmitActions(const std::vector<Ref<RunnerAction>>& Actions) static constexpr uint64_t SubmitWarnThresholdMs = 500; auto SubmitToRunner = [&](int RunnerIndex) { + auto& Runner = Runners[RunnerIndex]; + Runner->m_LastSubmitStats.Reset(); + Stopwatch Timer; - PerRunnerResults[RunnerIndex] = Runners[RunnerIndex]->SubmitActions(PerRunnerActions[RunnerIndex]); + PerRunnerResults[RunnerIndex] = Runner->SubmitActions(PerRunnerActions[RunnerIndex]); uint64_t ElapsedMs = Timer.GetElapsedTimeMs(); if (ElapsedMs >= SubmitWarnThresholdMs) { - ZEN_WARN("submit of {} actions to '{}' took {}ms", + size_t Attachments = Runner->m_LastSubmitStats.TotalAttachments.load(std::memory_order_relaxed); + uint64_t AttachmentBytes = Runner->m_LastSubmitStats.TotalAttachmentBytes.load(std::memory_order_relaxed); + + ZEN_WARN("submit of {} actions ({} attachments, {}) to '{}' took {}ms", PerRunnerActions[RunnerIndex].size(), - Runners[RunnerIndex]->GetDisplayName(), + Attachments, + NiceBytes(AttachmentBytes), + Runner->GetDisplayName(), ElapsedMs); } }; |