torch.compiler.is_compiling

torch.compiler.is_compiling()

指示图是否作为 torch.compile() 或 torch.export() 的一部分执行/追踪。

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

• torch._dynamo.external_utils.is_compiling()

• torch._utils.is_compiling()

示例：

>>> def forward(self, x):
>>>     if not torch.compiler.is_compiling():
>>>        ...逻辑 that is not needed in a compiled/traced graph...
>>>
>>>     ...rest of the function...

Return type

bool