Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Will Pytorch-DirectML support automatic mixed precision in the future? #192

Open
foemre opened this issue Jan 4, 2022 · 1 comment
Open

Will Pytorch-DirectML support automatic mixed precision in the future? #192

foemre opened this issue Jan 4, 2022 · 1 comment
Labels
pytorch-directml Issues in PyTorch when using its DirectML backend

Comments

@foemre
Copy link

foemre commented Jan 4, 2022

I am currently in the process of converting my code into pytorch-directml. Automatic mixed precision would be a great feature to have. If this is not possible in the near future, can someone help me with the usage of torch.dml ?
@jstoecker jstoecker added the pytorch-directml Issues in PyTorch when using its DirectML backend label Jan 10, 2022
@smk2007
Copy link
Member

smk2007 commented May 18, 2022

Hi, and thanks for the suggestion. Currently we are focusing on implementing missing operators and enabling models broadly on DirectML. We will look into additional features like mixed precision in the future as well.

@reid3333 reid3333 mentioned this issue Feb 2, 2023
Open
1 task
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
pytorch-directml Issues in PyTorch when using its DirectML backend
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@jstoecker @smk2007 @foemre