From e0962f4c374f4cbf78ad27fd2391f976c1a2050d Mon Sep 17 00:00:00 2001
From: Jiho Choi <jihochoi@google.com>
Date: Thu, 18 Jun 2020 13:35:31 -0700
Subject: [PATCH] Remove the old grouping rule for PJRT.
PiperOrigin-RevId: 317170383
Change-Id: I96973a2d2cd0ca1fc786bc7500deb6b4fedd0534
---
tensorflow/core/profiler/utils/group_events.cc | 5 +----
tensorflow/core/profiler/utils/xplane_schema.cc | 4 ----
tensorflow/core/profiler/utils/xplane_schema.h | 3 ---
3 files changed, 1 insertion(+), 11 deletions(-)
diff --git a/tensorflow/core/profiler/utils/group_events.cc b/tensorflow/core/profiler/utils/group_events.cc
index be8dd506b0c..0772cff7b97 100644
--- a/tensorflow/core/profiler/utils/group_events.cc
+++ b/tensorflow/core/profiler/utils/group_events.cc
@@ -635,10 +635,7 @@ std::vector<InterThreadConnectInfo> CreateInterThreadConnectInfoList() {
{StatType::kStepId, StatType::kIterNum}},
{HostEventType::kKernelLaunch,
HostEventType::kKernelExecute,
- {StatType::kCorrelationId}},
- {HostEventType::kLocalExecutableExecuteOnLocalDevice,
- HostEventType::kLocalExecutableExecute,
- {StatType::kRunId}}};
+ {StatType::kCorrelationId}}};
return connect_info_list;
}
diff --git a/tensorflow/core/profiler/utils/xplane_schema.cc b/tensorflow/core/profiler/utils/xplane_schema.cc
index be53dcbdc01..5ca8326d72c 100644
--- a/tensorflow/core/profiler/utils/xplane_schema.cc
+++ b/tensorflow/core/profiler/utils/xplane_schema.cc
@@ -91,10 +91,6 @@ const HostEventTypeMap& GetHostEventTypeMap() {
{"WhileOp-StartBody", kWhileOpStartBody},
{"ForOp", kForOp},
{"PartitionedCallOp", kPartitionedCallOp},
- // XLA related.
- {"LocalExecutable::ExecuteOnLocalDevices",
- kLocalExecutableExecuteOnLocalDevice},
- {"LocalExecutable::Execute", kLocalExecutableExecute},
// tf.data related.
{"IteratorGetNextOp::DoCompute", kIteratorGetNextOp},
{"IteratorGetNextAsOptionalOp::DoCompute", kIteratorGetNextAsOptionalOp},
diff --git a/tensorflow/core/profiler/utils/xplane_schema.h b/tensorflow/core/profiler/utils/xplane_schema.h
index a31814cef06..41774deaa59 100644
--- a/tensorflow/core/profiler/utils/xplane_schema.h
+++ b/tensorflow/core/profiler/utils/xplane_schema.h
@@ -81,9 +81,6 @@ enum HostEventType {
kWhileOpStartBody,
kForOp,
kPartitionedCallOp,
- // XLA related.
- kLocalExecutableExecuteOnLocalDevice,
- kLocalExecutableExecute,
// tf.data related.
kIteratorGetNextOp,
kIteratorGetNextAsOptionalOp,