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

single out host plane merge function to reuse.

Browse Source 
PiperOrigin-RevId: 338123012
Change-Id: Ib7d4dfb1fab09156c917a0b6465f88048f6cbb05
This commit is contained in:
A. Unique TensorFlower 2020-10-20 13:15:50 -07:00 committed by TensorFlower Gardener
parent 5c90e46f35
commit 169cbde464
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/core/profiler/convert/post_process_single_host_xplane.cc
View File

@ -22,10 +22,7 @@ limitations under the License.
namespace tensorflow {
namespace profiler {
void PostProcessSingleHostXSpace(XSpace* space, uint64 start_time_ns) {
VLOG(3) << "Post processing local profiler XSpace.";
// Post processing the collected XSpace without hold profiler lock.
// 1. Merge plane of host events with plane of CUPTI driver api.
void MergeHostPlanes(XSpace* space) {
const XPlane* cupti_driver_api_plane =
FindPlaneWithName(*space, kCuptiDriverApiPlaneName);
const XPlane* python_tracer_plane =
@ -49,6 +46,13 @@ void PostProcessSingleHostXSpace(XSpace* space, uint64 start_time_ns) {
RemovePlaneWithName(space, kPythonTracerPlaneName);
}
}
}
void PostProcessSingleHostXSpace(XSpace* space, uint64 start_time_ns) {
VLOG(3) << "Post processing local profiler XSpace.";
// Post processing the collected XSpace without hold profiler lock.
// 1. Merge plane of host events with plane of CUPTI driver api.
MergeHostPlanes(space);
// 2. Normalize all timestamps by shifting timeline to profiling start time.
// NOTE: this have to be done before sorting XSpace due to timestamp overflow.

3
tensorflow/core/profiler/convert/post_process_single_host_xplane.h
View File

@ -21,6 +21,9 @@ limitations under the License.
namespace tensorflow {
namespace profiler {
// Merges XPlanes generated by TraceMe, CUPTI API trace and Python tracer.
void MergeHostPlanes(XSpace* space);
// Post process XSpaces collected locally from multiple profilers.
void PostProcessSingleHostXSpace(XSpace* space, uint64 start_time_ns);