pytorch/benchmarks/dynamo/microbenchmarks
Gabriel Ferns 254293b777 Add flag _metrics_log_runtime to disable runtime metric logging by default (#153506)
https://github.com/pytorch/pytorch/pull/152708 expanded support of `get_estimated_runtime` to many more types of `SchedulerNodes`. This caused an increase in compile time because we're always calling `get_estimated_runtime` to populate the metrics table. This PR adds a flag for this logging, which reduces the instruction count by 8%. Long term, we should probably merge metrics.py with TORCH_LOGS/tlparse (suggestion from @xmfan).

Update: added support for TORCH_LOGS for the metrics logging.

Test Plan:
mm_loop.py and many existing tests cover.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153506
Approved by: https://github.com/eellison
2025-05-22 01:02:11 +00:00
..
operator_inp_logs
__init__.py
analyze_templates.py [BE] Format .ci/ / .github/ / benchmarks/ / functorch/ / tools/ / torchgen/ with ruff format (#132577) 2024-10-11 18:30:26 +00:00
bench_mm_fusion.py Add flag _metrics_log_runtime to disable runtime metric logging by default (#153506) 2025-05-22 01:02:11 +00:00
benchmark_helper.py
cache_debug_microbenchmarks.py [BE] fix ruff rule E226: add missing whitespace around operator in f-strings (#144415) 2025-01-08 21:55:00 +00:00
cache_hit_microbenchmarks.py Add a microbechmark for cache read path (#137607) 2024-10-10 16:36:18 +00:00
dynamo_guard_eval.py Rename cache limit to recompile limit in configs (#143709) 2024-12-22 10:03:57 +00:00
dynamo_microbenchmarks.py
fx_microbenchmarks.py [BE] fix ruff rule E226: add missing whitespace around operator in f-strings (#144415) 2025-01-08 21:55:00 +00:00
inductor_bmm.py
inductor_cpu_atomic.py
inductor_mm.py [BC breaking] move benchmarking + prefer inductor path (#132827) 2024-08-08 00:47:45 +00:00
matmul_relu.py
microbench.py
model.py
operator_inp_utils.py [BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546) 2025-02-27 20:46:16 +00:00
operatorbench.py [inductor] Benchmark Halide in operatorbench.py (#136809) 2024-09-28 19:26:04 +00:00
overheads.py [BE] fix ruff rule E226: add missing whitespace around operator in f-strings (#144415) 2025-01-08 21:55:00 +00:00
tensor_layout_mini_benchmark.py [BC breaking] move benchmarking + prefer inductor path (#132827) 2024-08-08 00:47:45 +00:00
utils.py