llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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.
pull/798/head
Maksim Levental 2022-04-25 14:34:32 -05:00 committed by GitHub
parent 6b3d0b7e7a
commit 693f79a2b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/torch_mlir/eager_mode/torch_mlir_dispatch.py
View File

@ -63,7 +63,7 @@ def normalize_args_kwargs(target: Callable, args: Tuple[Any], kwargs: Dict[str,
} }
new_args_and_kwargs = normalize_function( 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" assert new_args_and_kwargs, "Couldn't normalize args and kwargs"
new_args, new_kwargs = new_args_and_kwargs new_args, new_kwargs = new_args_and_kwargs

1
python/torch_mlir_e2e_test/test_suite/__init__.py
View File

@ -15,6 +15,7 @@ COMMON_TORCH_MLIR_LOWERING_XFAILS = {
"ConvolutionModule1D_basic", "ConvolutionModule1D_basic",
"MaxPool2dWith3dInputModule_basic", "MaxPool2dWith3dInputModule_basic",
"MaxPool2dWithIndicesWith3dInputModule_basic", "MaxPool2dWithIndicesWith3dInputModule_basic",
"ConstantPad2dStaticModule_basic",
} }
def register_all_tests(): def register_all_tests():