pytorch
3afc802c - [Static Runtime] Add Metadata to ProcessedNode depending upon the op type (#79961)

Commit
2 years ago
[Static Runtime] Add Metadata to ProcessedNode depending upon the op type (#79961) Summary: - ProcessedNodeMetadata class wraps the possible metadata for ProcessedNode. Depending upon the nature of op, processedNode can have one of the below possibilities of metadata: 1. prim::If/prim::Loop ops contains block_runners_ as their metadata 2. prim::fork op contains TaskLauncher (std::function) responsible for execution of forked subgraph Differential Revision: D37320704 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79961 Approved by: https://github.com/mikeiovine
Committer
Parents
Loading