torch.compiler.is_compiling#

torch.compiler.is_compiling()[source]#

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

Note that there are 2 other related flags that should deprecated eventually:

• torch._dynamo.external_utils.is_compiling()

• torch._utils.is_compiling()

Example:

>>>defforward(self,x):>>>ifnottorch.compiler.is_compiling():>>>pass# ...logic that is not needed in a compiled/traced graph...>>>>>># ...rest of the function...

Return type

bool