frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
17ab99463a62d3bde2cdbdd420b62784d15e5eef
pytorch/docs/source/cuda.tunable.md
Justin Silver f3e6c8e834 Fix #155016 for Docathon - convert rst to markdown (#155198) 
Used [rst2myst tool](https://rst-to-myst.readthedocs.io/en/latest/)

One note is that "Created On" and "Last Updated On" banner doesn't show in the markdown files... I'm not sure if that's just an artifact of my local build though.

Fixes #155016

Docs comparison (check out the 'new' whenever docs build)

1. cuda ([old](https://docs.pytorch.org/docs/main/cuda.html) vs. [new](https://docs-preview.pytorch.org/pytorch/pytorch/155198/cuda.html))
2. cuda.tunable ([old](https://docs.pytorch.org/docs/main/cuda.tunable.html) vs. [new](https://docs-preview.pytorch.org/pytorch/pytorch/155198/cuda.tunable.html))
3. leave cudnn_persistent_rnn.rst as is because it's reused in docstrings
4. cudnn_rnn_determinism.rst as is because it's reused in docstrings.
5. data ([old](https://docs.pytorch.org/docs/main/data.html) vs. [new](https://docs-preview.pytorch.org/pytorch/pytorch/155198/data.html))

Pull Request resolved: https://github.com/pytorch/pytorch/pull/155198
Approved by: https://github.com/albanD, https://github.com/svekars
2025-06-13 20:24:34 +00:00

1.3 KiB
Raw Blame History 

.. currentmodule:: torch.cuda.tunable

TunableOp

Overview

.. automodule:: torch.cuda.tunable

API Reference

.. autofunction:: enable

.. autofunction:: is_enabled

.. autofunction:: tuning_enable

.. autofunction:: tuning_is_enabled

.. autofunction:: record_untuned_enable

.. autofunction:: record_untuned_is_enabled

.. autofunction:: set_max_tuning_duration

.. autofunction:: get_max_tuning_duration

.. autofunction:: set_max_tuning_iterations

.. autofunction:: get_max_tuning_iterations

.. autofunction:: set_filename

.. autofunction:: get_filename

.. autofunction:: get_results

.. autofunction:: get_validators

.. autofunction:: write_file_on_exit

.. autofunction:: write_file

.. autofunction:: read_file

.. autofunction:: tune_gemm_in_file

.. autofunction:: mgpu_tune_gemm_in_file

.. autofunction:: set_rotating_buffer_size

.. autofunction:: get_rotating_buffer_size