search

intel / intel-extension-for-pytorch

A Python package for extending the official PyTorch that can easily obtain performance on Intel platform
Apache License 2.0
1.45k stars 220 forks source link

nvidia/amd gpu support for ipex #638

Closed bopeng1234 closed 1 week ago

bopeng1234 commented 1 month ago

Describe the issue

Hi Team,

from oneDNN/oneMKL link, both of them experimental support nvidia/amd GPU.

so, is there any plan to support nvidia/amd GPU in framework side, I mean ipex?

This question is similar like #346.

thanks! best regards.

xiguiw commented 1 month ago

@bopeng1234 No. At framework level, IPEX do NOT support nvidia/amd GPU.

MOLOjl commented 4 weeks ago

@xiguiw Hi, I'm considering to modify ipex to support nvidia/amd GPU, just use it to call onednn operaters(cudnn, miopen fuctions at the backend) on nvidia/amd GPU, is this easy? or tons of works I'm facing?

xiguiw commented 3 weeks ago

@MOLOjl Nvidia and AMD have their framework. so IPEX is NOT designed to do this. As I konw, on one did this before. Personally, if you just port some OP one a model, it's should be fine. But if you want to it works well, I don't think it's a easy thing personally.

xiguiw commented 1 week ago

close as no more action required.