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

Open source core_id_to_details map for adding host names in pod viewer.

Browse Source 
PiperOrigin-RevId: 336933370
Change-Id: I3f3dfa55359eace4590107696bc2cfde47c1bdcc
This commit is contained in:
A. Unique TensorFlower 2020-10-13 12:39:39 -07:00 committed by TensorFlower Gardener
parent 81a36b3bb3
commit 902cccc799
2 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/profiler/convert/op_stats_combiner.cc
View File

@ -157,6 +157,10 @@ void CombineOpStats(
// Combine tf-function stats. // Combine tf-function stats.
CombineTfFunctionDb(src.tf_function_db(), dst->mutable_tf_function_db()); CombineTfFunctionDb(src.tf_function_db(), dst->mutable_tf_function_db());
// Combine the mapping from core ID to details.
CombineCoreIdMap(src_host_id, src.core_id_to_details(),
dst->mutable_core_id_to_details());
} }
} // namespace } // namespace

13
tensorflow/core/profiler/protobuf/op_stats.proto
View File

@ -90,6 +90,17 @@ message RunEnvironment {
uint32 host_trace_level = 12; uint32 host_trace_level = 12;
} }
// Next ID: 7
message CoreDetails {
string hostname = 1;
uint32 device_ordinal = 2; // unique within host, TPU core only
uint32 core_num = 3; // unique within chip per core type
uint32 local_chip_id = 4; // unique within host
uint32 global_chip_id = 5; // unique within mesh
uint32 global_core_id = 6; // unique within mesh, TPU core only
}
// Next ID: 12
// Operator Statistics. // Operator Statistics.
message OpStats { message OpStats {
// The database for the op metrics collected from the host over the entire // The database for the op metrics collected from the host over the entire
@ -110,6 +121,8 @@ message OpStats {
KernelStatsDb kernel_stats_db = 6; KernelStatsDb kernel_stats_db = 6;
// Statistics for all tf-functions. // Statistics for all tf-functions.
TfFunctionDb tf_function_db = 8; TfFunctionDb tf_function_db = 8;
// A map from core ID to details.
map<uint32, CoreDetails> core_id_to_details = 11;
// Error and warning messages for diagnosing profiling issues. // Error and warning messages for diagnosing profiling issues.
Diagnostics diagnostics = 9; Diagnostics diagnostics = 9;
reserved 7; reserved 7;