diff --git a/sdk/TARGETS b/sdk/TARGETS new file mode 100644 index 00000000000..10fc8859c14 --- /dev/null +++ b/sdk/TARGETS @@ -0,0 +1,11 @@ +load("@fbcode_macros//build_defs:python_library.bzl", "python_library") + +oncall("executorch") + +python_library( + name = "lib", + srcs = ["__init__.py"], + deps = [ + "//executorch/sdk/etrecord:etrecord", + ], +) diff --git a/sdk/etdb/TARGETS b/sdk/inspector/TARGETS similarity index 100% rename from sdk/etdb/TARGETS rename to sdk/inspector/TARGETS diff --git a/sdk/etdb/_inspector_utils.py b/sdk/inspector/_inspector_utils.py similarity index 100% rename from sdk/etdb/_inspector_utils.py rename to sdk/inspector/_inspector_utils.py diff --git a/sdk/etdb/inspector.py b/sdk/inspector/inspector.py similarity index 99% rename from sdk/etdb/inspector.py rename to sdk/inspector/inspector.py index 1ff251d7114..2ba7556e6ee 100644 --- a/sdk/etdb/inspector.py +++ b/sdk/inspector/inspector.py @@ -26,14 +26,14 @@ from executorch.exir import ExportedProgram from executorch.sdk.edir.et_schema import OperatorGraphWithStats, OperatorNode -from executorch.sdk.etdb._inspector_utils import ( +from executorch.sdk.etdump.schema_flatcc import ETDumpFlatCC, ProfileEvent +from executorch.sdk.inspector._inspector_utils import ( create_debug_handle_to_op_node_mapping, EDGE_DIALECT_GRAPH_KEY, gen_etdump_object, gen_etrecord_object, gen_graphs_from_etrecord, ) -from executorch.sdk.etdump.schema_flatcc import ETDumpFlatCC, ProfileEvent from tabulate import tabulate diff --git a/sdk/etdb/tests/TARGETS b/sdk/inspector/tests/TARGETS similarity index 84% rename from sdk/etdb/tests/TARGETS rename to sdk/inspector/tests/TARGETS index f38f743dd88..898d49203fe 100644 --- a/sdk/etdb/tests/TARGETS +++ b/sdk/inspector/tests/TARGETS @@ -8,9 +8,9 @@ python_unittest( deps = [ "//executorch/exir:lib", "//executorch/sdk/edir:et_schema", - "//executorch/sdk/etdb:inspector", "//executorch/sdk/etrecord:etrecord", "//executorch/sdk/etrecord/tests:etrecord_test_library", + "//executorch/sdk/inspector:inspector", ], ) @@ -18,8 +18,8 @@ python_unittest( name = "event_blocks_test", srcs = ["event_blocks_test.py"], deps = [ - "//executorch/sdk/etdb:inspector", "//executorch/sdk/etdump:schema_flatcc", + "//executorch/sdk/inspector:inspector", ], ) @@ -28,8 +28,8 @@ python_unittest( srcs = ["inspector_utils_test.py"], deps = [ "//executorch/sdk/edir:et_schema", - "//executorch/sdk/etdb:inspector_utils", "//executorch/sdk/etrecord:etrecord", "//executorch/sdk/etrecord/tests:etrecord_test_library", + "//executorch/sdk/inspector:inspector_utils", ], ) diff --git a/sdk/etdb/tests/event_blocks_test.py b/sdk/inspector/tests/event_blocks_test.py similarity index 99% rename from sdk/etdb/tests/event_blocks_test.py rename to sdk/inspector/tests/event_blocks_test.py index a6aae7e7d4f..a333a35b961 100644 --- a/sdk/etdb/tests/event_blocks_test.py +++ b/sdk/inspector/tests/event_blocks_test.py @@ -9,14 +9,14 @@ from typing import List, Optional, Tuple, Union import executorch.sdk.etdump.schema_flatcc as flatcc -from executorch.sdk.etdb.inspector import ( +from executorch.sdk.etdump.schema_flatcc import ETDumpFlatCC, ProfileEvent +from executorch.sdk.inspector.inspector import ( DelegateMetadata, Event, EventBlock, PerfData, ProfileEventSignature, ) -from executorch.sdk.etdump.schema_flatcc import ETDumpFlatCC, ProfileEvent class TestEventBlock(unittest.TestCase): diff --git a/sdk/etdb/tests/inspector_test.py b/sdk/inspector/tests/inspector_test.py similarity index 98% rename from sdk/etdb/tests/inspector_test.py rename to sdk/inspector/tests/inspector_test.py index a17bc925682..494489a3b36 100644 --- a/sdk/etdb/tests/inspector_test.py +++ b/sdk/inspector/tests/inspector_test.py @@ -16,13 +16,13 @@ from executorch.exir import ExportedProgram from executorch.sdk.edir.et_schema import OperatorNode - -from executorch.sdk.etdb import inspector - -from executorch.sdk.etdb.inspector import Event, EventBlock, Inspector, PerfData from executorch.sdk.etrecord import generate_etrecord, parse_etrecord from executorch.sdk.etrecord.tests.etrecord_test import TestETRecord +from executorch.sdk.inspector import inspector + +from executorch.sdk.inspector.inspector import Event, EventBlock, Inspector, PerfData + OP_TYPE = "aten::add" EVENT_BLOCK_NAME = "block_0" diff --git a/sdk/etdb/tests/inspector_utils_test.py b/sdk/inspector/tests/inspector_utils_test.py similarity index 98% rename from sdk/etdb/tests/inspector_utils_test.py rename to sdk/inspector/tests/inspector_utils_test.py index a963c47abd5..00247155998 100644 --- a/sdk/etdb/tests/inspector_utils_test.py +++ b/sdk/inspector/tests/inspector_utils_test.py @@ -14,14 +14,14 @@ OperatorNode, ValueNode, ) -from executorch.sdk.etdb._inspector_utils import ( +from executorch.sdk.etrecord import generate_etrecord, parse_etrecord + +from executorch.sdk.etrecord.tests.etrecord_test import TestETRecord +from executorch.sdk.inspector._inspector_utils import ( create_debug_handle_to_op_node_mapping, EDGE_DIALECT_GRAPH_KEY, gen_graphs_from_etrecord, ) -from executorch.sdk.etrecord import generate_etrecord, parse_etrecord - -from executorch.sdk.etrecord.tests.etrecord_test import TestETRecord class TestInspectorUtils(unittest.TestCase):