Skip to content

Commit 8a02d77

Browse files
authored
Add a few xfails. (huggingface#477)
1 parent 1cdf301 commit 8a02d77

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

tank/test_models.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -407,6 +407,28 @@ def test_module(self, dynamic, device, config):
407407
and device == "cpu"
408408
):
409409
pytest.xfail(reason="https://github.com/nod-ai/SHARK/issues/463")
410+
if (
411+
config["model_name"]
412+
in [
413+
"facebook/convnext-tiny-224",
414+
"squeezenet1_0",
415+
]
416+
and device == "rocm"
417+
):
418+
pytest.xfail(
419+
reason="iree-compile buffer limit issue: https://github.com/nod-ai/SHARK/issues/475"
420+
)
421+
if (
422+
config["model_name"]
423+
in [
424+
"funnel-transformer/small",
425+
"mobilenet_v3_small",
426+
]
427+
and device == "rocm"
428+
):
429+
pytest.xfail(
430+
reason="Numerics issues: https://github.com/nod-ai/SHARK/issues/476"
431+
)
410432
if config["framework"] == "tf" and dynamic == True:
411433
pytest.skip(
412434
reason="Dynamic shapes not supported for this framework."

0 commit comments

Comments
 (0)