pytorch
517ea26e - [deploy] Make load_library a no-op inside a package (#58933)

Commit
3 years ago
[deploy] Make load_library a no-op inside a package (#58933) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58933 **Summary** This commit makes load_library calls no-ops inside packages run with deploy. Libraries containing custom C++ operators and classes are statically linked in C++ and don't need to be loaded. This commit takes advantage of the fact that sys.executable is set to torch_deploy in deploy and uses that to exit early in load_library if the program is running inside deploy. **Test Plan** This commit adds a test to `generate_examples`/`test_deploy` that packages and runs a function that calls `load_library`. The library doesn't exist, but that's okay because the function should be a no-op anyway. Test Plan: Imported from OSS Reviewed By: Lilyjjo Differential Revision: D28687159 Pulled By: SplitInfinity fbshipit-source-id: 4a61fc636698e44f204334e338c5ce35257e7ae2
Author
Meghan Lele
Parents
Loading