pytorch
9d9cc675 - Make torch.library.define consistent with the new APIs (#111307)

Commit
1 year ago
Make torch.library.define consistent with the new APIs (#111307) This PR introduces a new overload of torch.library.define. Like impl_abstract, and our plans for the rest of the torch.library APIs, we allow it to accept an optional library object to tie the lifetime of the op definition to. Test Plan: - new tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/111307 Approved by: https://github.com/soulitzer, https://github.com/ezyang
Author
Committer
Parents
Loading