benchmark

033e172b - Grab bag of (mostly) typing improvements (#158075)

Grab bag of (mostly) typing improvements (#158075) Summary: Collects some scattershot improvements made while attempting to enable training for AOTInductor. Non-typing changes are: 1. Swapping a few custom searches for the output node in an FX graph for calling `graph.output_node()`. 2. Removing two unused parameters from `torch.export._unlift._unlift`. 3. Switching handles to constants in `cpp_wrapper_cpu` to use C++ references for memory efficiency. 4. Cleaning out unused, unexported imports from `torch/export/__init__.py`, and adding one missing export to `__all__`. X-link: https://github.com/pytorch/pytorch/pull/158075 Approved by: https://github.com/Skylion007 Reviewed By: ZainRizvi Differential Revision: D78691998 fbshipit-source-id: cdd51fe27cef89786ac7728775c103f5c11fcb1b