torch.compiler.is_dynamo_compiling

torch.compiler.is_dynamo_compiling() [исходный код]

Указывает, трассируется ли граф с помощью TorchDynamo.

Это более строгий флаг, чем is_compiling(): он принимает значение True, только если используется TorchDynamo.

Пример:

>>> def forward(self, x):
>>>     if not torch.compiler.is_dynamo_compiling():
>>>        pass # ...logic that is not needed in a TorchDynamo-traced graph...
>>>
>>>     # ...rest of the function...

Тип возвращаемого значения:

bool