pytorch
ee1d375b - [FSDP] Add fast path for `NO_SHARD` `clip_grad_norm_()` (#89137)

Commit
2 years ago
[FSDP] Add fast path for `NO_SHARD` `clip_grad_norm_()` (#89137) Pull Request resolved: https://github.com/pytorch/pytorch/pull/89137 Approved by: https://github.com/rohan-varma
Author
Committer
Parents
Loading