docs/source/benchmark_utils.md
.. role:: hidden
:class: hidden-section
.. automodule:: torch.utils.benchmark
.. currentmodule:: torch.utils.benchmark
.. autoclass:: Timer
:members:
.. autoclass:: Measurement
:members:
.. autoclass:: CallgrindStats
:members:
.. autoclass:: FunctionCounts
:members:
.. autoclass:: Compare
:members:
.. currentmodule:: torch.utils.benchmark.examples.op_benchmark
.. autofunction:: assert_dicts_equal
.. currentmodule:: torch.utils.benchmark.utils.fuzzer
.. autofunction:: prod
.. currentmodule:: torch.utils.benchmark.utils.common
.. autofunction:: select_unit
.. autofunction:: trim_sigfig
.. currentmodule:: torch.utils.benchmark.utils.compile
.. autofunction:: bench_all
.. autofunction:: benchmark_compile
.. py:module:: torch.utils.benchmark.examples
.. automodule:: torch.utils.benchmark.examples.spectral_ops_fuzz_test
.. currentmodule:: torch.utils.benchmark.examples.spectral_ops_fuzz_test
.. autosummary::
:toctree: generated
:nosignatures:
run_benchmark
.. py:module:: torch.utils.benchmark.op_fuzzers
.. py:module:: torch.utils.benchmark.utils
.. py:module:: torch.utils.benchmark.utils.valgrind_wrapper