optimum icon indicating copy to clipboard operation
optimum copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

Experimental DirectML support via torch-directml

Open kazssym opened this issue 1 year ago • 0 comments

What does this PR do?

This PR adds experimental DirectML support via torch-directml, which is still in preview and lacks several PyTorch functions such as microsoft/DirectML#449.

If you are interested in this PR, please leave comments below.

Before submitting

  • [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • [ ] Did you make sure to update the documentation with your changes?
  • [ ] Did you write any new necessary tests?

kazssym avatar Feb 19 '24 14:02 kazssym