Allow expanding and collapsing in aten::view

- Supports cases where the view op expands and collapses dims simulataneously. This does not handle the case where it is neither expanding nor collapsing (e.g. [2, 3] -> [3, 2]) - Additionally fixes a previous bug with adding 1-sized dims on both sides of a tensor with aten.view
llvm · Jul 20, 2022 · 244ba2e · 244ba2e
1 parent 7f08169
commit 244ba2e
Show file tree

Hide file tree

Showing 3 changed files with 354 additions and 150 deletions.
diff --git a/e2e_testing/torchscript/xfail_sets.py b/e2e_testing/torchscript/xfail_sets.py
@@ -145,6 +145,8 @@
     "ViewExpandOnesModule_basic",
     "ViewExpandOnesBeforeAndAfterModule_basic",
     "ViewExpandOnesMiddleModule_basic",
+    "ViewExpandCollapseModule_basic",
+    "ViewExpandCollapseWithOnesModule_basic",
     "ViewCollapseInferredDimModule_basic",
     "ViewExpandInferredDimModule_basic",
     "ViewNoChangeStaticModule_basic",