Skip to content

Commit 943e44b

Browse files
authored
Add evaluation cases (#245)
* Add evaluation cases * mypy
1 parent 7158b2f commit 943e44b

File tree

4 files changed

+194
-138
lines changed

4 files changed

+194
-138
lines changed

_doc/status/exported_program_dynamic.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ with different options. This steps happens before converting into ONNX.
5353
for exporter in (
5454
"export-strict",
5555
"export-nostrict",
56+
"export-nostrict-oblivious",
5657
"export-nostrict-decall",
5758
"export-tracing",
5859
):

_unittests/ut_torch_export_patches/test_eval.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,22 @@ def test_run_exporter_custom_nested_cond(self):
8686
dynamic=False,
8787
)
8888

89+
def test_run_exporter_dimension0(self):
90+
evaluation(
91+
cases="ExportWithDimension0",
92+
exporters="export-nostrict-oblivious",
93+
quiet=False,
94+
dynamic=True,
95+
)
96+
97+
def test_run_exporter_dimension1(self):
98+
evaluation(
99+
cases="ExportWithDimension1",
100+
exporters="export-nostrict-oblivious",
101+
quiet=False,
102+
dynamic=True,
103+
)
104+
89105

90106
if __name__ == "__main__":
91107
unittest.main(verbosity=2)

0 commit comments

Comments
 (0)