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 deprecated custom forward method #151

Merged
merged 2 commits into from
Aug 31, 2024
Merged

Fix deprecated custom forward method #151

merged 2 commits into from
Aug 31, 2024

Conversation

ziw-liu
Copy link
Collaborator

@ziw-liu ziw-liu commented Aug 30, 2024

torch.cuda.amp.custom_fwd() is deprecated in favor of torch.amp.custom_fwd().

@ziw-liu ziw-liu added the maintenance Maintenance work label Aug 30, 2024
@ziw-liu ziw-liu requested a review from edyoshikun August 30, 2024 14:15
@ziw-liu ziw-liu added this to the v0.2.1 milestone Aug 30, 2024
@ziw-liu ziw-liu marked this pull request as ready for review August 31, 2024 00:12
@ziw-liu ziw-liu merged commit c5b9ab6 into main Aug 31, 2024
4 checks passed
@ziw-liu ziw-liu deleted the fix-amp-deprecation branch August 31, 2024 00:13
@edyoshikun edyoshikun mentioned this pull request Sep 26, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
maintenance Maintenance work
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants