enable several cases on XPU (#37516)

* enable several cases on XPU

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* Update tests/test_modeling_common.py

Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>

* fix style

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

---------

Signed-off-by: YAO Matrix <matrix.yao@intel.com>
Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
This commit is contained in:
Yao Matrix
2025-04-16 17:01:04 +08:00
committed by GitHub
parent 5ab7a7c640
commit 33f6c5a5c8
4 changed files with 10 additions and 8 deletions

View File

@@ -22,6 +22,7 @@ from transformers.testing_utils import (
require_bitsandbytes,
require_torch,
require_torch_accelerator,
require_torch_multi_accelerator,
require_torch_multi_gpu,
slow,
torch_device,
@@ -517,7 +518,7 @@ class FalconMambaIntegrationTests(unittest.TestCase):
self.assertListEqual(out, EXPECTED_OUTPUT)
@require_torch_multi_gpu
@require_torch_multi_accelerator
def test_training_kernel(self):
model_id = "tiiuae/falcon-mamba-7b"