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

Tells the compiler frontend (Dynamo) to skip symbolic introspection of the function and instead directly write it to the graph when encountered.

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.

is_compiling

Indicates whether a graph is executed/traced as part of torch.compile() or torch.export().

is_dynamo_compiling

Indicates whether a graph is traced via TorchDynamo.

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