pytorch
9ae3f394 - Add remote_module logging to the __new__ method. (#68035)

Commit
3 years ago
Add remote_module logging to the __new__ method. (#68035) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68035 RemoteModule is sometimes created using object.__new__ (ex: init_from_module_rref), in this case the logging in the __init__ method would not pick this up. As a result, adding a `__new__` method to RemoteModule to log all usages appropriately. ghstack-source-id: 142762019 Test Plan: waitforbuildbot Reviewed By: vipannalla Differential Revision: D32263978 fbshipit-source-id: a95ab0bb5d0836da8fe6333c41593af164b008d9
Author
Parents
Loading