From e195703a6f1a74910977549bf3fe257031676eca Mon Sep 17 00:00:00 2001 From: Masahiro Hiramori Date: Tue, 28 Nov 2023 19:38:54 +0900 Subject: [PATCH] enable `test_transformer` --- tests/python/frontend/pytorch/test_forward.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py index 14dd3ae225f03..5c54314907ab1 100644 --- a/tests/python/frontend/pytorch/test_forward.py +++ b/tests/python/frontend/pytorch/test_forward.py @@ -4776,7 +4776,6 @@ def test_fn(x, mask): verify_model(test_fn, [inp.to(torch.float64), inp > 0.5]) -@pytest.mark.skip(reason="unsupported op: 'aten::scaled_dot_product_attention'") def test_transformer(): """test_transformer""" model = torch.nn.Transformer(d_model=256, nhead=8, num_encoder_layers=6, num_decoder_layers=6)