Skip to content

Commit f697b3f

Browse files
authored
enable 2 types of case on XPU (#37198)
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>
1 parent 2099287 commit f697b3f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_modeling_common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1978,7 +1978,7 @@ def test_resize_tokens_embeddings_with_deepspeed(self):
19781978
self.test_resize_tokens_embeddings()
19791979

19801980
@require_deepspeed
1981-
@require_torch_multi_gpu
1981+
@require_torch_multi_accelerator
19821982
def test_resize_tokens_embeddings_with_deepspeed_multi_gpu(self):
19831983
ds_config = {
19841984
"zero_optimization": {
@@ -2084,7 +2084,7 @@ def test_resize_embeddings_untied_with_deepspeed(self):
20842084
self.test_resize_embeddings_untied()
20852085

20862086
@require_deepspeed
2087-
@require_torch_multi_gpu
2087+
@require_torch_multi_accelerator
20882088
def test_resize_embeddings_untied_with_deepspeed_multi_gpu(self):
20892089
ds_config = {
20902090
"zero_optimization": {

0 commit comments

Comments
 (0)