Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Improve torch::jit::as_{module,object} performance (pytorch#84399)
This caches the import of `torch.jit.ScriptModule`, `torch.ScriptObject` and `torch.jit.RecursiveScriptClass`. I measure a ~0.8 us performance uplift locally when calling a `torch.ops` function with a `ScriptObject` argument. Pull Request resolved: pytorch#84399 Approved by: https://github.com/ezyang
- Loading branch information