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

Change the prefix of xprof arguments from "$" to "_".

Browse Source 
PiperOrigin-RevId: 316166741
Change-Id: I3a83f550755e6c34a179781d61ab6c24b03cca13
This commit is contained in:
Jiho Choi 2020-06-12 13:29:55 -07:00 committed by TensorFlower Gardener
parent cd6a929e30
commit 287a60c12a
5 changed files with 12 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/common_runtime/direct_session.cc
View File

@ -511,11 +511,11 @@ Status DirectSession::RunInternal(
model_metadata.version());
return profiler::TraceMeEncode("SessionRun",
{{"id", step_id},
{"$r", 1} /*root_event*/,
{"_r", 1} /*root_event*/,
{"model_id", model_id}});
} else {
return profiler::TraceMeEncode(
"SessionRun", {{"id", step_id}, {"$r", 1} /*root_event*/});
"SessionRun", {{"id", step_id}, {"_r", 1} /*root_event*/});
}
},
profiler::ContextType::kTfExecutor, step_id,

2
tensorflow/core/common_runtime/eager/kernel_and_device.cc
View File

@ -387,7 +387,7 @@ void KernelAndDeviceFunc::RunAsync(
// To TraceMeConsumers in ExecutorState::Process/Finish.
[&] {
return profiler::TraceMeEncode(
"FunctionRun", {{"id", opts->step_id}, {"$r", 1} /*root_event*/});
"FunctionRun", {{"id", opts->step_id}, {"_r", 1} /*root_event*/});
},
profiler::ContextType::kTfExecutor, opts->step_id,
profiler::TraceMeLevel::kInfo);

2
tensorflow/core/distributed_runtime/graph_mgr.cc
View File

@ -424,7 +424,7 @@ void GraphMgr::ExecuteAsync(const string& handle, const int64 step_id,
// To TraceMeConsumers in ExecutorState::Process/Finish or RunGraphDone.
[step_id] {
return profiler::TraceMeEncode(
"RunGraph", {{"id", step_id}, {"$r", 1} /*root_event*/});
"RunGraph", {{"id", step_id}, {"_r", 1} /*root_event*/});
},
profiler::ContextType::kTfExecutor, step_id,
profiler::TraceMeLevel::kInfo);

6
tensorflow/core/profiler/lib/connected_traceme.h
View File

@ -88,8 +88,7 @@ class TraceMeProducer {
trace_me_.AppendMetadata([&] {
context_id_ =
context_id.has_value() ? *context_id : TraceMe::NewActivityId();
return TraceMeEncode(
{{"$pt", static_cast<int>(context_type)}, {"$p", context_id_}});
return TraceMeEncode({{"_pt", context_type}, {"_p", context_id_}});
});
}
@ -107,8 +106,7 @@ class TraceMeConsumer {
int level = 2)
: trace_me_(name, level) {
trace_me_.AppendMetadata([&] {
return TraceMeEncode(
{{"$ct", static_cast<int>(context_type)}, {"$c", context_id}});
return TraceMeEncode({{"_ct", context_type}, {"_c", context_id}});
});
}

12
tensorflow/core/profiler/utils/xplane_schema.cc
View File

@ -148,12 +148,12 @@ const StatTypeMap& GetStatTypeMap() {
{"kpi_name", kKpiName},
{"kpi_value", kKpiValue},
// XPlane semantics related.
{"$pt", kProducerType},
{"$ct", kConsumerType},
{"$p", kProducerId},
{"$c", kConsumerId},
{"$r", kIsRoot},
{"$a", kIsAsync},
{"_pt", kProducerType},
{"_ct", kConsumerType},
{"_p", kProducerId},
{"_c", kConsumerId},
{"_r", kIsRoot},
{"_a", kIsAsync},
// Device trace arguments.
{"device_id", kDeviceId},
{"context_id", kContextId},