torch.compiler API reference
For a quick overview of torch.compiler
, see torch.compiler.
See |
|
This function clears all compilation caches and restores the system to its initial state. |
|
Tells the compiler frontend (Dynamo) to skip symbolic introspection of the function and instead directly write it to the graph when encountered. |
|
Register a polyfill handler for a function, usually a C function from the C extension, to be used in place of the original function when inlining the original function in the graph. |
|
This function is used to mark a function fn as having a constant result. |
|
Return valid strings that can be passed to torch.compile(..., backend="name"). |
|
This function provides a decorator to disable compilation on a function It also provides the option of recursively disabling called functions |
|
Set the current stance of the compiler. |
|
Indicates that a new iteration of inference or training is about to begin. |
|
Indicates whether a graph is executed/traced as part of torch.compile() or torch.export(). |
|
Indicates whether a graph is traced via TorchDynamo. |