pytorch

c371c3ab - [rpc][profiler] add a test case to verify record_function context manager works (#34511)

[rpc][profiler] add a test case to verify record_function context manager works (#34511) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34511 With https://github.com/pytorch/pytorch/pull/34122/files, issues with using record_function context manager and profiling RPCs were fixed. This adds a test case to verify that we can use RPC with the `record_function` decorator. ghstack-source-id: 100109932 Test Plan: Unit test change Differential Revision: D20352242 fbshipit-source-id: d6429e4352ad3b8d874dc0f27b23ecb6202e6b2b