Back to Pytorch

Benchmark Utils - torch.utils.benchmark

docs/source/benchmark_utils.md

2.13.01.6 KB
Original Source
{eval-rst}
.. role:: hidden
    :class: hidden-section

Benchmark Utils - torch.utils.benchmark

{eval-rst}
.. automodule:: torch.utils.benchmark
{eval-rst}
.. currentmodule:: torch.utils.benchmark
{eval-rst}
.. autoclass:: Timer
    :members:
{eval-rst}
.. autoclass:: Measurement
    :members:
{eval-rst}
.. autoclass:: CallgrindStats
    :members:
{eval-rst}
.. autoclass:: FunctionCounts
    :members:
{eval-rst}
.. autoclass:: Compare
    :members:
{eval-rst}
.. currentmodule:: torch.utils.benchmark.examples.op_benchmark

.. autofunction:: assert_dicts_equal
{eval-rst}
.. currentmodule:: torch.utils.benchmark.utils.fuzzer

.. autofunction:: prod
{eval-rst}
.. currentmodule:: torch.utils.benchmark.utils.common

.. autofunction:: select_unit

.. autofunction:: trim_sigfig
{eval-rst}
.. currentmodule:: torch.utils.benchmark.utils.compile

.. autofunction:: bench_all

.. autofunction:: benchmark_compile

torch.utils.benchmark.examples

{eval-rst}
.. py:module:: torch.utils.benchmark.examples

torch.utils.benchmark.examples.spectral_ops_fuzz_test

{eval-rst}
.. automodule:: torch.utils.benchmark.examples.spectral_ops_fuzz_test
{eval-rst}
.. currentmodule:: torch.utils.benchmark.examples.spectral_ops_fuzz_test
{eval-rst}
.. autosummary::
    :toctree: generated
    :nosignatures:

    run_benchmark
{eval-rst}
.. py:module:: torch.utils.benchmark.op_fuzzers
{eval-rst}
.. py:module:: torch.utils.benchmark.utils
{eval-rst}
.. py:module:: torch.utils.benchmark.utils.valgrind_wrapper