This is an automated email from the ASF dual-hosted git repository.
tlopex pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 83db389868 [Relax][PyTorch] Enable decomposition in all tests (#18464)
83db389868 is described below
commit 83db389868a3c582763dc09f79c85a5a739dec77
Author: Guan-Ming (Wesley) Chiu <[email protected]>
AuthorDate: Tue Nov 18 13:07:01 2025 +0800
[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
---
tests/python/relax/test_frontend_from_exported_program.py | 11 +++++++++--
1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/tests/python/relax/test_frontend_from_exported_program.py
b/tests/python/relax/test_frontend_from_exported_program.py
index 001df64815..87022a2d7d 100644
--- a/tests/python/relax/test_frontend_from_exported_program.py
+++ b/tests/python/relax/test_frontend_from_exported_program.py
@@ -6435,7 +6435,14 @@ def test_dynamic_shape():
batch = torch.export.Dim("batch")
dynamic_shapes = {"x1": {0: batch}, "x2": {0: batch}}
- verify_model(DynamicModel(), example_args, {}, Expected,
dynamic_shapes=dynamic_shapes)
+ verify_model(
+ DynamicModel(),
+ example_args,
+ {},
+ Expected,
+ dynamic_shapes=dynamic_shapes,
+ run_ep_decomposition=True,
+ )
def test_broadcast_to():
@@ -6919,7 +6926,7 @@ def test_tensor_none_tuple():
R.output(gv)
return gv
- verify_model(TensorNoneModel(), example_args, {}, Expected)
+ verify_model(TensorNoneModel(), example_args, {}, Expected,
run_ep_decomposition=True)
def test_gru():
