torch.compiler.allow_in_graph

torch.compiler.allow_in_graph(fn) [source]

Настраивает, какие функции будут включены в сгенерированную граф. Это обходит все средства интроспекции символического Python-кода в пользу непосредственного его записи в граф. Если fn — список или кортеж вызываемых объектов, то он рекурсивно применяет allow_in_graph() к каждой функции и возвращает новый список или кортеж, содержащий изменённые функции.

Параметры

fn – вызываемый объект, представляющий функцию, которая должна быть включена в граф.

Предупреждение

allow_in_graph() полностью пропускает TorchDynamo для декорированной функции, пропуская все проверки безопасности TorchDynamo (разрывы графа, обработка замыканий и т. д.). Поэтому необходимо быть очень осторожным при использовании allow_in_graph(), так как подсистемы, такие как AOT Autograd, полагаются на torchdynamo. Небрежность может привести к проблемам с корректностью и трудно отлаживаемым ошибкам.