On this page
torch.compiler API reference
For a quick overview of torch.compiler
, see torch.compiler.
compile |
See |
reset |
This function clears all compilation caches and restores the system to its initial state. |
allow_in_graph |
Customize which functions compilation will include in the generated graph. |
assume_constant_result |
This function is used to mark a function |
list_backends |
Return valid strings that can be passed to |
disable |
This function provides both a decorator and a context manager to disable compilation on a function It also provides the option of recursively disabling called functions |
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/torch.compiler_api.html