Benchmark RecordFunction overhead on some models (#40952)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40952
Adding a benchmark to measure RecordFunction overhead,
currently on resnet50 and lstm models
Test Plan:
python benchmarks/record_function_benchmark/record_function_bench.py
Benchmarking RecordFunction overhead for lstm_jit
Running warmup... finished
Running 100 iterations with RecordFunction... finished
N = 100, avg. time: 251.970 ms, stddev: 39.348 ms
Running 100 iterations without RecordFunction... finished
N = 100, avg. time: 232.828 ms, stddev: 24.556 ms
Reviewed By: dzhulgakov
Differential Revision: D22368357
Pulled By: ilia-cher
fbshipit-source-id: bff4f4e0e06fb80fdfcf85966c2468e48ed7bc98