pytorch
afb4914c - Align torch.library.impl with the new torch.library style (#111308)

Commit
1 year ago
Align torch.library.impl with the new torch.library style (#111308) We add a new overload to torch.library.impl that accepts an optional Library arg. If provided, the lifetime of the registration will be tied to the Library arg, otherwise, it will live forever. Test Plan: - existing and new tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/111308 Approved by: https://github.com/soulitzer ghstack dependencies: #111307
Author
Committer
Parents
Loading