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:
>>> def forward(self, x): >>> if not torch.compiler.is_compiling(): >>> pass # ...logic that is not needed in a compiled/traced graph... >>> >>> # ...rest of the function...
- Return type