Shortcuts

torch.compiler API reference

For a quick overview of torch.compiler, see torch.compiler.

compile

See torch.compile() for details on the arguments for this function.

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 fn as having a constant result.

list_backends

Return valid strings that can be passed to torch.compile(..., backend="name").

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

cudagraph_mark_step_begin

Indicates that a new iteration of inference or training is about to begin.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources