pytorch

65a37923 - [Static Runtime] Exception handling during fork subgraph execution (#79292)

[Static Runtime] Exception handling during fork subgraph execution (#79292) Summary: - Exception handling was not performed in forked subgraph execution - forked subgraph runtime can throw runtime exception. Future returned by prim::fork needs to handle exceptions so that aten::wait handles it. Test Plan: local test cases: - buck test caffe2/benchmarks/static_runtime/fb:test_fb_operators - buck test mode/opt caffe2/benchmarks/static_runtime:static_runtime_cpptest - buck test mode/opt caffe2/test:static_runtime Async execution of the subgraph is tested by adding pytorch profiler hooks on the StaticRuntime execution via below code. Async execution in threadpool is verfiied by checking trace with profile(activities=[ProfilerActivity.CPU]) as prof: static_runtime_module(inputs) prof.export_chrome_trace("trace.json") Differential Revision: D37072493 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79292 Approved by: https://github.com/mikeiovine