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.

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

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

jondeaton
Copy link

@jondeaton jondeaton commented Sep 6, 2024

Fixes warning about deprecation of torch.get_autocast_gpu_dtype

@timmoon10
Copy link
Collaborator

/te-ci pytorch

Copy link
Collaborator

@timmoon10 timmoon10 left a 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?

Copy link
Member

@ksivaman ksivaman left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@jondeaton
Copy link
Author

Can you sign your commit to get past the DCO check?

Yep sure thing, I think I've signed off properly now.

@ksivaman
Copy link
Member

ksivaman commented Sep 6, 2024

Looks like the first commit is still not signed off, could you try to sign-off all commits? Thanks! @jondeaton

jondeaton and others added 2 commits September 6, 2024 14:00
Signed-off-by: Jon Deaton <jon.deaton@evolutionaryscale.ai>
for more information, see https://pre-commit.ci

Signed-off-by: Jon Deaton <jon.deaton@evolutionaryscale.ai>
@jondeaton
Copy link
Author

Yeah, sorry about that - new to the process. I think I've signed off on all of them now?

Copy link
Member

@ksivaman ksivaman left a 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!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants