intel-extension-for-pytorch icon indicating copy to clipboard operation
intel-extension-for-pytorch copied to clipboard
verified publisher icon intel

Feature request: upstream changes to pytorch

Open Pandapip1 opened this issue 1 year ago • 2 comments

Easier said than done, but: I'd really like it if you could upstream these changes to the main pytorch project. This comes with a number of advantages:

  1. It allows Intel Arc cards to be used for all pytorch applications
  2. It shifts the maintenance burden onto the pytorch project

Pandapip1 avatar Feb 27 '24 17:02 Pandapip1

Hi, this is a great suggestion. Actually, we do work very closely with the main PyTorch developers to upstream a lot of our optimizations. Right now, it is limited to CPU optimizations. In general, the intention of the extension is to provide the most up-to-date features and hardware targeted performance optimizations. A lot of this will eventually be upstreamed in later releases.

kminhta avatar Feb 27 '24 22:02 kminhta

Great! I'll keep this issue open for now, since there are non-upstreamed optimizations, but I'm glad to hear that this is on the roadmap! Go Intel Arc!

Pandapip1 avatar Feb 29 '24 16:02 Pandapip1