Skip to content

Commit

Permalink
[Pytorch] Disable failing tests on AArch64 (apache#12255)
Browse files Browse the repository at this point in the history
Change-Id: I234a4486a4a5ffac03d478e65cbfcee3efe30df1
  • Loading branch information
NicolaLancellotti authored Aug 2, 2022
1 parent d10a7d1 commit 19644bc
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions tests/python/frontend/pytorch/test_forward.py
100644 → 100755
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
"""Unit tests for various models and operators"""
from contextlib import suppress
import os
import platform
import sys
from time import time

Expand Down Expand Up @@ -4060,6 +4061,10 @@ def test_forward_pretrained_bert_base_uncased():
print("TVM top-1 id: {}, token: {}".format(tvm_pred_idx, tvm_pred_token))


@pytest.mark.skipif(
platform.machine() == "aarch64",
reason="Currently failing on AArch64",
)
def test_convert_torch_script_with_input_types():
def model_fn(x, y):
x = x.to(dtype=torch.int32)
Expand Down

0 comments on commit 19644bc

Please sign in to comment.