pytorch

b8019cee - ns for fx: make input logging work for multi-node subgraphs (#54326)

ns for fx: make input logging work for multi-node subgraphs (#54326) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54326 Fixes unshadowed activation input logging for subgraphs where start_node does not equal end_node. In detail: * instead of passing around a single list of nodes, pass around a list of nodes to instrument inputs, and a list of nodes to instrument outputs. This way we can handle multi-node subgraphs properly, and we also keep the subgraph instance definition out of the public APIs. * add a test case Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_linear_fp16_activations ``` Imported from OSS Reviewed By: hx89 Differential Revision: D27190138 fbshipit-source-id: 58e2377c1c128baaf3b760c1ad29098fb21f53d3