Skip to content

Commit 1cc834e

Browse files
Jack-Khuufacebook-github-bot
authored andcommitted
Cleaning et_schema of InferenceRun for OSS (#565)
Summary: These are concepts that aren't useful in the OSS world. Associated tests were also moved - Moved out InferenceRun, OperatorGraphWithStats, ExportedETOperatorGraph for internal - Changed FXOperatorGraph to extending OperatorGraph instead of OperatorGraphWithStats - Tests for EDIR (to be renamed) moved to internal since they are heavily coupled with InferenceRun Reviewed By: tarun292 Differential Revision: D49839120
1 parent 80ee5f6 commit 1cc834e

31 files changed

+104
-709
lines changed

sdk/edir/TARGETS

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,8 @@ python_library(
1717
"et_schema.py",
1818
],
1919
deps = [
20-
"fbsource//third-party/pypi/numpy:numpy",
2120
":base_schema",
2221
"//caffe2:torch",
2322
"//executorch/exir:lib",
24-
"//executorch/exir:schema",
25-
"//executorch/exir/_serialize:lib",
26-
"//executorch/sdk/etdump:schema",
27-
"//executorch/sdk/etdump:serialize",
2823
],
2924
)

sdk/edir/et_schema.py

Lines changed: 2 additions & 589 deletions
Large diffs are not rendered by default.

sdk/edir/tests/fixtures/composite_delegate_module_fx_graph_aten_dialect.txt

Lines changed: 0 additions & 3 deletions
This file was deleted.

sdk/edir/tests/fixtures/composite_delegate_module_fx_graph_edge_dialect.txt

Lines changed: 0 additions & 3 deletions
This file was deleted.

sdk/edir/tests/fixtures/composite_delegate_module_fx_graph_et_dialect.txt

Lines changed: 0 additions & 3 deletions
This file was deleted.

sdk/edir/tests/fixtures/two_linear_module_fx_graph_aten_dialect.txt

Lines changed: 0 additions & 3 deletions
This file was deleted.

sdk/edir/tests/fixtures/two_linear_module_fx_graph_edge_dialect.txt

Lines changed: 0 additions & 3 deletions
This file was deleted.

sdk/edir/tests/fixtures/two_linear_module_fx_graph_et_dialect.txt

Lines changed: 0 additions & 3 deletions
This file was deleted.

sdk/edir/tests/TARGETS renamed to sdk/fb/tests/TARGETS

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@ python_unittest(
1515
"//caffe2:torch",
1616
"//executorch/exir:lib",
1717
"//executorch/exir:lowered_backend_module",
18-
"//executorch/sdk/edir:base_schema",
1918
"//executorch/sdk/edir:et_schema",
2019
"//executorch/sdk/etdump:schema",
20+
"//executorch/sdk/fb:et_schema",
2121
],
2222
)
2323

@@ -31,9 +31,9 @@ python_library(
3131
"//caffe2:torch",
3232
"//executorch/exir:lib",
3333
"//executorch/exir:lowered_backend_module",
34-
"//executorch/sdk/edir:base_schema",
3534
"//executorch/sdk/edir:et_schema",
3635
"//executorch/sdk/etdump:schema",
36+
"//executorch/sdk/fb:et_schema",
3737
],
3838
)
3939

@@ -42,14 +42,14 @@ python_binary(
4242
srcs = [
4343
"exported_op_graph_test.py",
4444
],
45-
main_module = "executorch.sdk.edir.tests.exported_op_graph_test",
45+
main_module = "executorch.sdk.fb.tests.exported_op_graph_test",
4646
deps = [
4747
"fbsource//third-party/pypi/parameterized:parameterized",
4848
"//caffe2:torch",
4949
"//executorch/exir:lib",
5050
"//executorch/exir:lowered_backend_module",
51-
"//executorch/sdk/edir:base_schema",
5251
"//executorch/sdk/edir:et_schema",
5352
"//executorch/sdk/etdump:schema",
53+
"//executorch/sdk/fb:et_schema",
5454
],
5555
)

0 commit comments

Comments
 (0)