transformers
[tests] make `test_model_parallelism` device-agnostic
#30844

Merged

[tests] make `test_model_parallelism` device-agnostic #30844

amyeroberts merged 4 commits into huggingface:main from faaany:parallelism

enable on xpu

f99e2ae0

fix style

ff0b240c

ydshieh assigned

ydshieh 1 year ago

ydshieh commented on 2024-05-16

ydshieh requested a review from

ydshieh 1 year ago

ydshieh approved these changes on 2024-05-23

SunMarc approved these changes on 2024-05-23

add comment and mps

f975ab83

Merge branch 'huggingface:main' into parallelism

7531f064

amyeroberts approved these changes on 2024-05-24

amyeroberts merged 04c7c176 into main 1 year ago

Reviewers

SunMarc

ydshieh

amyeroberts

Assignees

ydshieh

Labels

None yet

Milestone

No milestone