pytorch
f125bd2c - Support torch.ScriptObject in torch::jit::as_object (#84398)

Commit
3 years ago
Support torch.ScriptObject in torch::jit::as_object (#84398) When a torchbind class is returned from an operator, it has the class `torch.ScriptObject`, yet the `torch.ops` interface checks against `torch.jit.RecursiveScriptClass` or else falls back to a much slower path that doesn't return the original c++ object. On my machine I see a 2 us performance improvement when calling a `torch.ops` function with a `ScriptObject` argument. Pull Request resolved: https://github.com/pytorch/pytorch/pull/84398 Approved by: https://github.com/ezyang
Author
Committer
Parents
Loading