Go
pytorch
78a053ba
- [activation checkpointing] Add default autocast keys to functional rng wrappers (#107934)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
1 year ago
[activation checkpointing] Add default autocast keys to functional rng wrappers (#107934) Pull Request resolved: https://github.com/pytorch/pytorch/pull/107934 Approved by: https://github.com/xw285cornell
Author
anijain2305
Committer
pytorchmergebot
Parents
3992450e
Loading