SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[tests] skip tests for xpu (#33553)

Browse Source 
* enable

* fix

* add xpu skip

* add marker

* skip for xpu

* add more

* add one more
This commit is contained in:
Fanli Lin
2024-09-20 02:28:04 +08:00
committed by GitHub
parent f111d5b783
commit b87755aa6d
6 changed files with 26 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/models/layoutlmv2/test_modeling_layoutlmv2.py
View File

@@ -16,7 +16,14 @@
import unittest
from transformers.testing_utils import require_detectron2, require_torch, require_torch_multi_gpu, slow, torch_device
from transformers.testing_utils import (
require_detectron2,
require_non_xpu,
require_torch,
require_torch_multi_gpu,
slow,
torch_device,
)
from transformers.utils import is_detectron2_available, is_torch_available
from ...test_configuration_common import ConfigTester
@@ -251,6 +258,7 @@ class LayoutLMv2ModelTester:
return config, inputs_dict
@require_non_xpu
@require_torch
@require_detectron2
class LayoutLMv2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):