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

Triage e2e IREE failures for npcetest. 

ResNet works with static shapes. (our test is not static though).

All tests blocked on https://github.com/google/iree/issues/6629
pull/272/head
Sean Silva 2021-08-05 12:00:38 -07:00
parent c464cb107f
commit f71845ea75
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frontends/pytorch/e2e_testing/torchscript/main.py
View File

@ -41,7 +41,7 @@ def _get_argparse():
if IREE_ENABLED: if IREE_ENABLED:
config_choices += ['iree'] config_choices += ['iree']
parser = argparse.ArgumentParser(description='Run torchscript e2e tests.') parser = argparse.ArgumentParser(description='Run torchscript e2e tests.')
parser.add_argument('--config', parser.add_argument('-c', '--config',
choices=config_choices, choices=config_choices,
default='refbackend', default='refbackend',
help=f''' help=f'''

9
frontends/pytorch/e2e_testing/torchscript/xfail_sets.py
View File

@ -21,10 +21,9 @@ _common_npcomp_lowering_xfails = {
XFAIL_SETS['refbackend'] = _common_npcomp_lowering_xfails XFAIL_SETS['refbackend'] = _common_npcomp_lowering_xfails
XFAIL_SETS['iree'] = _common_npcomp_lowering_xfails | { XFAIL_SETS['iree'] = _common_npcomp_lowering_xfails | {
#https://reviews.llvm.org/D106658 to reach iree release # https://github.com/google/iree/issues/6629
'MaxPool2dModule_basic', 'Conv2dWithPaddingModule_basic',
'Conv2dWithPaddingDilationStrideModule_basic', 'Conv2dWithPaddingDilationStrideModule_basic',
#https://github.com/google/iree/issues/6420 'ResNet18Module_basic',
'FlattenDynamicModule_basic', 'MaxPool2dModule_basic',
'ResNet18Module_basic'
} }