github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

xfail for cpu models with tensor shape inf error (#512)

Browse Source
This commit is contained in:
Phaneesh Barwaria
2022-11-25 03:42:04 +05:30
committed by GitHub
parent 88937fcb2f
commit 04f20070d1
1 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tank/test_models.py
View File

@@ -448,6 +448,30 @@ class SharkModuleTest(unittest.TestCase):
pytest.xfail(
reason="Numerics issues: https://github.com/nod-ai/SHARK/issues/489"
)
if (
config["model_name"]
in [
"microsoft/resnet-50",
"alexnet",
"resnet101",
"resnet18",
"resnet50",
"wide_resnet50_2",
]
and device
in [
"cpu",
"cuda",
]
and config["framework"] == "torch"
):
pytest.xfail(
reason="tensor dimension issue: https://github.com/nod-ai/SHARK/issues/511"
)
if config["model_name"] == "squeezenet1_0" and device == "cuda":
pytest.xfail(
reason="tensor dimension issue: https://github.com/nod-ai/SHARK/issues/511"
)
if config["framework"] == "tf" and dynamic == True:
pytest.skip(
reason="Dynamic shapes not supported for this framework."