pytorch

b763666f - [JIT] Subgraph utils: add an optional vmap argument to the API to allow retrieving value mappings. (#43235)

[JIT] Subgraph utils: add an optional vmap argument to the API to allow retrieving value mappings. (#43235) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43235 This functionality is needed when we want to not lose track of nodes/values as we merge and unmerge them into other nodes. For instance, if we have a side data structure with some meta information about values or nodes, this new functionality would allow to keep that metadata up to date after merging and unmerging nodes. Differential Revision: D23202648 Test Plan: Imported from OSS Reviewed By: eellison Pulled By: ZolotukhinM fbshipit-source-id: 350d21a5d462454166f8a61b51d833551c49fcc9