-
Notifications
You must be signed in to change notification settings - Fork 366
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.
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
Fix autocast deprecation warning. #1167
base: main
Are you sure you want to change the base?
Conversation
/te-ci pytorch |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the fix, this looks good to me. Can you sign your commit to get past the DCO check?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
f750a06
to
252787e
Compare
Yep sure thing, I think I've signed off properly now. |
Looks like the first commit is still not signed off, could you try to sign-off all commits? Thanks! @jondeaton |
Signed-off-by: Jon Deaton <[email protected]>
for more information, see https://pre-commit.ci Signed-off-by: Jon Deaton <[email protected]>
252787e
to
9abde8f
Compare
Yeah, sorry about that - new to the process. I think I've signed off on all of them now? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
On second thought, we had this change previously in a PR and ended up reverting it here. Since this API is a relatively new addition to pytorch, we would want it to be accompanied by a pytorch version check so that versions 2.3.1 or older aren't broken!
Fixes warning about deprecation of
torch.get_autocast_gpu_dtype