Skip to content

Commit 1538910

Browse files
angelayifacebook-github-bot
authored andcommitted
Update exir.pass_base to use export.pass_base
Summary: X-link: pytorch/pytorch#106647 Also fixed T159713621 Reviewed By: tugsbayasgalan Differential Revision: D48068293 fbshipit-source-id: 873fdce0eb230cde1eef7c0e8dd7734cf58a75cf
1 parent 6fab1e5 commit 1538910

File tree

8 files changed

+37
-520
lines changed

8 files changed

+37
-520
lines changed

backends/test/test_backends_lifted.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1067,8 +1067,6 @@ def f(x, y):
10671067
false_lowered[0][1].original_module.graph_module.code
10681068
)
10691069

1070-
# TODO(T159713621): Fix this test
1071-
@unittest.expectedFailure
10721070
def test_partition_with_map(self) -> None:
10731071
def map_fn(x, y):
10741072
x = x - y
@@ -1116,8 +1114,6 @@ def f(xs, y):
11161114

11171115
self.assertTrue(torch.allclose(orig_res, new_res[0]))
11181116

1119-
# TODO(T159713621): Fixing this test
1120-
@unittest.expectedFailure
11211117
def test_partition_with_nested_control_flow(self) -> None:
11221118
"""
11231119
Partitions the add and mul ops, including the ones inside the submodules

exir/TARGETS

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -232,8 +232,6 @@ python_library(
232232
"//caffe2:torch",
233233
"//caffe2/functorch:functorch_src",
234234
"//executorch/exir/dialects/edge:lib",
235-
"//executorch/exir/pass_infra:node_metadata",
236-
"//executorch/exir/pass_infra:proxy_value",
237235
],
238236
)
239237

0 commit comments

Comments
 (0)