Back to Pytorch

TunableOp

docs/source/cuda.tunable.md

2.11.01.2 KB
Original Source
{eval-rst}
.. currentmodule:: torch.cuda.tunable

TunableOp

Overview

{eval-rst}
.. automodule:: torch.cuda.tunable

API Reference

{eval-rst}
.. autofunction:: enable
{eval-rst}
.. autofunction:: is_enabled
{eval-rst}
.. autofunction:: tuning_enable
{eval-rst}
.. autofunction:: tuning_is_enabled
{eval-rst}
.. autofunction:: record_untuned_enable
{eval-rst}
.. autofunction:: record_untuned_is_enabled
{eval-rst}
.. autofunction:: set_max_tuning_duration
{eval-rst}
.. autofunction:: get_max_tuning_duration
{eval-rst}
.. autofunction:: set_max_tuning_iterations
{eval-rst}
.. autofunction:: get_max_tuning_iterations
{eval-rst}
.. autofunction:: set_filename
{eval-rst}
.. autofunction:: get_filename
{eval-rst}
.. autofunction:: get_results
{eval-rst}
.. autofunction:: get_validators
{eval-rst}
.. autofunction:: read_file
{eval-rst}
.. autofunction:: tune_gemm_in_file
{eval-rst}
.. autofunction:: mgpu_tune_gemm_in_file
{eval-rst}
.. autofunction:: set_rotating_buffer_size
{eval-rst}
.. autofunction:: get_rotating_buffer_size
{eval-rst}
.. autofunction:: set_numerical_check_tolerances