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

enable 2 types of case on XPU (#37198)

Browse Source 
enable 2 types of case on XPU 1. test_resize_tokens_embeddings_with_deepspeed_multi_gpu 2. test_resize_embeddings_untied_with_deepspeed_multi_gpu

Signed-off-by: YAO Matrix <matrix.yao@intel.com>
This commit is contained in:
Yao Matrix
2025-04-03 17:37:55 +08:00
committed by GitHub
parent 2099287a59
commit f697b3f824
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_modeling_common.py
View File

@@ -1978,7 +1978,7 @@ class ModelTesterMixin:
self.test_resize_tokens_embeddings()
@require_deepspeed
@require_torch_multi_gpu
@require_torch_multi_accelerator
def test_resize_tokens_embeddings_with_deepspeed_multi_gpu(self):
ds_config = {
"zero_optimization": {
@@ -2084,7 +2084,7 @@ class ModelTesterMixin:
self.test_resize_embeddings_untied()
@require_deepspeed
@require_torch_multi_gpu
@require_torch_multi_accelerator
def test_resize_embeddings_untied_with_deepspeed_multi_gpu(self):
ds_config = {
"zero_optimization": {