Skip to content

Commit 2075c2a

Browse files
committed
Update based on PR comments
1 parent c26bbd0 commit 2075c2a

File tree

4 files changed

+19
-9
lines changed

4 files changed

+19
-9
lines changed

devtools/etdump/tests/etdump_test.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -820,8 +820,10 @@ TEST_F(ProfilerETDumpTest, LogDelegateEvents) {
820820
const char* metadata = "test_metadata";
821821
etdump_gen[i]->log_profiling_delegate(
822822
nullptr, 278, 1, 2, metadata, strlen(metadata) + 1);
823-
EventTracerEntry entry = etdump_gen[i]->start_profiling_delegate(
824-
"test_event", kUnsetDelegateDebugIntId).get();
823+
Result<EventTracerEntry> res = etdump_gen[i]->start_profiling_delegate(
824+
"test_event", kUnsetDelegateDebugIntId);
825+
ASSERT_TRUE(res.ok());
826+
EventTracerEntry entry = res.get();
825827
EXPECT_NE(entry.delegate_event_id_type, DelegateDebugIdType::kNone);
826828
// Event 2
827829
etdump_gen[i]->end_profiling_delegate(

runtime/core/event_tracer.h

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -216,9 +216,13 @@ class EventTracer {
216216
* @param[in] delegate_debug_index The id of the delegate event. If string
217217
* based names are used by this delegate to identify ops executed in the
218218
* backend then kUnsetDebugHandle should be passed in here.
219-
* @return Returns an instance of EventTracerEntry which should be passed back
220-
* into the end_profiling_dele gate() call.
221-
* - An error code if an error occurs during logging.
219+
* @return Returns an Result<EventTracerEntry> instance which may contatain
220+
* a EventTracerEntry instance. The EventTracerEntry instance should be
221+
* passed back into the end_profiling_delegate() call.
222+
* - Result<EventTracerEntry>::ok() is true if the event tracer entry
223+
* was successfully created.
224+
* - Result<EventTracerEntry>::error() is true if an error occurs
225+
* during logging.
222226
*/
223227
virtual Result<EventTracerEntry> start_profiling_delegate(
224228
const char* name,

runtime/core/event_tracer_hooks_delegate.h

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,10 @@ inline EventTracerEntry event_tracer_start_profiling_delegate(
5353
DebugHandle delegate_debug_id) {
5454
#ifdef ET_EVENT_TRACER_ENABLED
5555
if (event_tracer) {
56-
return event_tracer->start_profiling_delegate(name, delegate_debug_id);
56+
Result<EventTracerEntry> res = event_tracer->start_profiling_delegate(
57+
name, delegate_debug_id);
58+
ET_CHECK_MSG(res.ok(), "Failed to start profiling delegate event");
59+
return res.get();
5760
}
5861
#else //! ET_EVENT_TRACER_ENABLED
5962
(void)name;
@@ -179,8 +182,9 @@ inline void event_tracer_log_output_delegate(
179182
std::is_same<T, executorch::aten::Tensor>::value ||
180183
std::is_same<T, ArrayRef<executorch::aten::Tensor>>::value,
181184
"Unsupported type for intermediate output");
182-
event_tracer->log_intermediate_output_delegate(
185+
Result<bool> res = event_tracer->log_intermediate_output_delegate(
183186
name, delegate_debug_id, output);
187+
ET_CHECK_MSG(res.ok(), "Failed to log intermediate output delegate event");
184188
}
185189
#else //! ET_EVENT_TRACER_ENABLED
186190
(void)name;

runtime/core/test/event_tracer_test.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ class DummyEventTracer : public EventTracer {
109109
(void)end_time;
110110
(void)metadata;
111111
(void)metadata_len;
112-
return Result<bool>(true);
112+
return true;
113113
}
114114

115115
virtual Result<bool> log_intermediate_output_delegate(
@@ -237,7 +237,7 @@ TEST(TestEventTracer, SimpleEventTracerTest) {
237237
*/
238238
void RunSimpleTracerTestDelegate(EventTracer* event_tracer) {
239239
EventTracerEntry event_tracer_entry = event_tracer_start_profiling_delegate(
240-
event_tracer, "test_event", kUnsetDelegateDebugIntId).get();
240+
event_tracer, "test_event", kUnsetDelegateDebugIntId);
241241
event_tracer_end_profiling_delegate(
242242
event_tracer, event_tracer_entry, nullptr);
243243
event_tracer_start_profiling_delegate(event_tracer, nullptr, 1);

0 commit comments

Comments
 (0)