Skip to content

Commit 83db389

Browse files
authored
[Relax][PyTorch] Enable decomposition in all tests (#18464)
## Why This is the last part of the migration. After this one, we could set `run_ep_decomposition` default to true in our test ## How Check and update the remaining tests
1 parent 8dc9f5f commit 83db389

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

tests/python/relax/test_frontend_from_exported_program.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6435,7 +6435,14 @@ def main(
64356435
batch = torch.export.Dim("batch")
64366436
dynamic_shapes = {"x1": {0: batch}, "x2": {0: batch}}
64376437

6438-
verify_model(DynamicModel(), example_args, {}, Expected, dynamic_shapes=dynamic_shapes)
6438+
verify_model(
6439+
DynamicModel(),
6440+
example_args,
6441+
{},
6442+
Expected,
6443+
dynamic_shapes=dynamic_shapes,
6444+
run_ep_decomposition=True,
6445+
)
64396446

64406447

64416448
def test_broadcast_to():
@@ -6919,7 +6926,7 @@ def main(
69196926
R.output(gv)
69206927
return gv
69216928

6922-
verify_model(TensorNoneModel(), example_args, {}, Expected)
6929+
verify_model(TensorNoneModel(), example_args, {}, Expected, run_ep_decomposition=True)
69236930

69246931

69256932
def test_gru():

0 commit comments

Comments
 (0)