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

[tf:tfrt] Improve tf_cpurt tracing by adding more Executable details

Browse Source 
PiperOrigin-RevId: 413413792
Change-Id: I72b53cb8066eee93a89a62553e5c2e05b177349b
This commit is contained in:
Eugene Zhulenev 2021-12-01 07:44:14 -08:00 committed by TensorFlower Gardener
parent cdc20a431c
commit 2d910c08e6
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/compiler/mlir/tfrt/jit/tf_cpurt_kernels.cc
View File

@ -508,8 +508,12 @@ static void ExecuteImpl(Executable& executable,
TraceMe trace_me([&] {
int64_t id = exec_ctx.request_ctx()->id();
absl::string_view name(executable.name().data(), executable.name().size());
return TraceMeEncode("tf_cpurt.Execute",
{{"id", id}, {"executable", name}});
return TraceMeEncode(
"tf_cpurt.Execute",
{{"id", id},
{"executable", name},
{"specialized", executable.specialized() ? "true" : "false"},
{"num_worker_threads", executable.num_worker_threads()}});
});
// Keep track of memory address to tensor mapping for result conversion.