Fix test fails due to upstream PyTorch change (llvm#793)

* Add to eager tests to xfail while they are fixed. Also XFAIL ConstantPad2dStaticModule_basic. * Fix test fails due to upstream PyTorch change.
gpetters94 · Apr 25, 2022 · 693f79a · 693f79a
1 parent 6b3d0b7
commit 693f79a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/python/torch_mlir/eager_mode/torch_mlir_dispatch.py b/python/torch_mlir/eager_mode/torch_mlir_dispatch.py
@@ -63,7 +63,7 @@ def normalize_args_kwargs(target: Callable, args: Tuple[Any], kwargs: Dict[str,
     }
 
     new_args_and_kwargs = normalize_function(
-        target, args, kwargs, arg_types, kwarg_types, normalize_to_only_use_kwargs=False
+        target.op, args, kwargs, arg_types, kwarg_types, normalize_to_only_use_kwargs=False
     )
     assert new_args_and_kwargs, "Couldn't normalize args and kwargs"
     new_args, new_kwargs = new_args_and_kwargs

diff --git a/python/torch_mlir_e2e_test/test_suite/__init__.py b/python/torch_mlir_e2e_test/test_suite/__init__.py
@@ -15,6 +15,7 @@
     "ConvolutionModule1D_basic",
     "MaxPool2dWith3dInputModule_basic",
     "MaxPool2dWithIndicesWith3dInputModule_basic",
+    "ConstantPad2dStaticModule_basic",
 }
 
 def register_all_tests():