Add knobs to set the number of profiling runs and bailout depth (#32735)
Summary:
Diagnostic API to simplify debugging and experiments.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/32735
Differential Revision: D19626708
Pulled By: Krovatkin
fbshipit-source-id: aa8c0da94d4559329fd7c8093329aea4e0271b6a