Skip to content

benchmark_utils.md

Latest commit

 

History

History
103 lines (75 loc) · 1.61 KB

benchmark_utils.md

File metadata and controls

103 lines (75 loc) · 1.61 KB
Raw
.. role:: hidden
    :class: hidden-section

Benchmark Utils - torch.utils.benchmark

.. 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

torch.utils.benchmark.examples

.. py:module:: torch.utils.benchmark.examples

torch.utils.benchmark.examples.spectral_ops_fuzz_test

.. 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