From 693f79a2b63e72d3b2af5f6528b0fe37ec6c394a Mon Sep 17 00:00:00 2001 From: Maksim Levental Date: Mon, 25 Apr 2022 14:34:32 -0500 Subject: [PATCH] Fix test fails due to upstream PyTorch change (#793) * Add to eager tests to xfail while they are fixed. Also XFAIL ConstantPad2dStaticModule_basic. * Fix test fails due to upstream PyTorch change. --- python/torch_mlir/eager_mode/torch_mlir_dispatch.py | 2 +- python/torch_mlir_e2e_test/test_suite/__init__.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/python/torch_mlir/eager_mode/torch_mlir_dispatch.py b/python/torch_mlir/eager_mode/torch_mlir_dispatch.py index ca4ac0828..b494dfe33 100644 --- 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 index 561be39f5..00affbcc0 100644 --- a/python/torch_mlir_e2e_test/test_suite/__init__.py +++ b/python/torch_mlir_e2e_test/test_suite/__init__.py @@ -15,6 +15,7 @@ COMMON_TORCH_MLIR_LOWERING_XFAILS = { "ConvolutionModule1D_basic", "MaxPool2dWith3dInputModule_basic", "MaxPool2dWithIndicesWith3dInputModule_basic", + "ConstantPad2dStaticModule_basic", } def register_all_tests():