Skip to content

Commit 04f2007

Browse files
authored
xfail for cpu models with tensor shape inf error (huggingface#512)
1 parent 88937fc commit 04f2007

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

tank/test_models.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,30 @@ def test_module(self, dynamic, device, config):
448448
pytest.xfail(
449449
reason="Numerics issues: https://github.com/nod-ai/SHARK/issues/489"
450450
)
451+
if (
452+
config["model_name"]
453+
in [
454+
"microsoft/resnet-50",
455+
"alexnet",
456+
"resnet101",
457+
"resnet18",
458+
"resnet50",
459+
"wide_resnet50_2",
460+
]
461+
and device
462+
in [
463+
"cpu",
464+
"cuda",
465+
]
466+
and config["framework"] == "torch"
467+
):
468+
pytest.xfail(
469+
reason="tensor dimension issue: https://github.com/nod-ai/SHARK/issues/511"
470+
)
471+
if config["model_name"] == "squeezenet1_0" and device == "cuda":
472+
pytest.xfail(
473+
reason="tensor dimension issue: https://github.com/nod-ai/SHARK/issues/511"
474+
)
451475
if config["framework"] == "tf" and dynamic == True:
452476
pytest.skip(
453477
reason="Dynamic shapes not supported for this framework."

0 commit comments

Comments
 (0)