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

View File

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

View File

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