[faaany]

What does this PR do?

There are many places in the task guides that only check the cuda device availability. This PR adds xpu besides cuda, but I am not sure whether my current change is the best one, because there are other ways to add xpu as, e.g.

• add xpu in the inline-comment
• use accelerate's get_backend API to get the underlying device

Just let me what is your favorite, @stevhliu Once agreed by you, I can submit further commits to replicate the change in other places.

[stevhliu]

Thanks for the suggestion!

I think using get_backend may be the easiest option for users, and this will also include other devices such as mps.

[faaany]

Thanks for the suggestion!

I think using get_backend may be the easiest option for users, and this will also include other devices such as mps.

thanks! code update. Is this okay for you? Then I can change to other places as well.

[stevhliu]

Very nice, thank you! 👍

[faaany]

Very nice, thank you! 👍

I only made one change: I removed GPU from the comment because GPU is a general term including CUDA, XPU, NPU. And I made the same change at other places as well. Pls have a review, thx a lot!

[stevhliu]

Fantastic, works for me!

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.