|
8 | 8 |
|
9 | 9 | #include <gtest/gtest.h> |
10 | 10 |
|
| 11 | +#include <executorch/runtime/core/array_ref.h> |
| 12 | +#include <executorch/runtime/core/evalue.h> |
11 | 13 | #include <executorch/runtime/core/event_tracer.h> |
12 | 14 | // Enable flag for test |
13 | 15 | #define ET_EVENT_TRACER_ENABLED |
14 | 16 | #include <executorch/runtime/core/event_tracer_hooks.h> |
15 | 17 | #include <executorch/runtime/core/event_tracer_hooks_delegate.h> |
16 | 18 |
|
17 | | -namespace torch { |
18 | | -namespace executor { |
19 | | - |
20 | | -using namespace internal; |
| 19 | +using exec_aten::Tensor; |
| 20 | +using executorch::runtime::AllocatorID; |
| 21 | +using executorch::runtime::ArrayRef; |
| 22 | +using executorch::runtime::ChainID; |
| 23 | +using executorch::runtime::DebugHandle; |
| 24 | +using executorch::runtime::EValue; |
| 25 | +using executorch::runtime::EventTracer; |
| 26 | +using executorch::runtime::EventTracerDebugLogLevel; |
| 27 | +using executorch::runtime::EventTracerEntry; |
| 28 | +using executorch::runtime::kUnsetChainId; |
| 29 | +using executorch::runtime::kUnsetDebugHandle; |
| 30 | +using executorch::runtime::LoggedEValueType; |
21 | 31 |
|
22 | 32 | class DummyEventTracer : public EventTracer { |
23 | 33 | public: |
@@ -159,6 +169,14 @@ class DummyEventTracer : public EventTracer { |
159 | 169 | * Exercise all the event_tracer API's for a basic sanity check. |
160 | 170 | */ |
161 | 171 | void RunSimpleTracerTest(EventTracer* event_tracer) { |
| 172 | + using executorch::runtime::internal::event_tracer_begin_profiling_event; |
| 173 | + using executorch::runtime::internal::event_tracer_create_event_block; |
| 174 | + using executorch::runtime::internal::event_tracer_end_profiling_event; |
| 175 | + using executorch::runtime::internal::event_tracer_track_allocation; |
| 176 | + using executorch::runtime::internal::event_tracer_track_allocator; |
| 177 | + using executorch::runtime::internal::EventTracerProfileInstructionScope; |
| 178 | + using executorch::runtime::internal::EventTracerProfileScope; |
| 179 | + |
162 | 180 | event_tracer_create_event_block(event_tracer, "ExampleEvent"); |
163 | 181 | event_tracer_create_event_block(event_tracer, "ExampleEvent"); |
164 | 182 | EventTracerEntry event_entry = |
@@ -216,6 +234,9 @@ TEST(TestEventTracer, SimpleEventTracerTestDelegate) { |
216 | 234 | } |
217 | 235 |
|
218 | 236 | TEST(TestEventTracer, SimpleEventTracerTestLogging) { |
| 237 | + using executorch::runtime::internal::event_tracer_log_evalue; |
| 238 | + using executorch::runtime::internal::event_tracer_log_evalue_output; |
| 239 | + |
219 | 240 | EValue test_eval(true); |
220 | 241 |
|
221 | 242 | { |
@@ -259,8 +280,5 @@ TEST(TestEventTracer, SimpleEventTracerTestLogging) { |
259 | 280 | event_tracer_log_evalue_output(nullptr, test_eval); |
260 | 281 | } |
261 | 282 |
|
262 | | -} // namespace executor |
263 | | -} // namespace torch |
264 | | - |
265 | | -// TODO : (T163645377) Add more test coverage to log and verify events passed |
266 | | -// into DummyTracer. |
| 283 | +// TODO(T163645377): Add more test coverage to log and verify events passed into |
| 284 | +// DummyTracer. |
0 commit comments