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.
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.