mmengine icon indicating copy to clipboard operation
mmengine copied to clipboard

[Feature] Support for XPU devices

Open iori2333 opened this issue 1 year ago • 1 comments

What is the feature?

According to PyTorch Documentation, PyTorch 2.5 will support Intel GPUs as compute backend (so-called xpu devices). I looked into source of mmengine and find that only minor changes are need to adopt XPU as backend.

Any other context?

No response

iori2333 avatar Oct 14 '24 15:10 iori2333

I made a simple implementation here. I got it tested with my Arc A770 using unmodified mmsegmentation and it worked fine.

iori2333 avatar Oct 14 '24 15:10 iori2333