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

Add KernelLaunch and KernelExecute event types.

Browse Source 
PiperOrigin-RevId: 291844655
Change-Id: I002aa2a71e35b60a502394a1a5da13ea73290a78
This commit is contained in:
Jiho Choi 2020-01-27 18:30:51 -08:00 committed by TensorFlower Gardener
parent 05445032af
commit d641d29ca5
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -77,6 +77,9 @@ const HostEventTypeMap& GetHostEventTypeMap() {
{"WhileOp-StartBody", kWhileOpStartBody}, {"WhileOp-StartBody", kWhileOpStartBody},
{"ForOp", kForOp}, {"ForOp", kForOp},
{"PartitionedCallOp", kPartitionedCallOp}, {"PartitionedCallOp", kPartitionedCallOp},
// GPU related.
{"KernelLaunch", kKernelLaunch},
{"KernelExecute", kKernelExecute},
}); });
DCHECK_EQ(host_event_type_map->size(), kNumHostEventTypes); DCHECK_EQ(host_event_type_map->size(), kNumHostEventTypes);
return *host_event_type_map; return *host_event_type_map;

5
tensorflow/core/profiler/utils/xplane_schema.h
View File

@ -69,7 +69,10 @@ enum HostEventType {
kWhileOpStartBody, kWhileOpStartBody,
kForOp, kForOp,
kPartitionedCallOp, kPartitionedCallOp,
kLastHostEventType = kPartitionedCallOp, // GPU related.
kKernelLaunch,
kKernelExecute,
kLastHostEventType = kKernelExecute,
}; };
enum StatType { enum StatType {