torch.compiler API справочник

Для быстрого обзора torch.compiler, см. torch.compiler.

`compile`	См. `torch.compile()` для подробностей об аргументах этой функции.
`reset`	Эта функция очищает все кэши компиляции и восстанавливает систему в исходное состояние.
`allow_in_graph`	Настройте, какие функции компиляция включит в сгенерированный граф.
`assume_constant_result`	Эта функция используется для пометки функции `fn` как имеющей постоянный результат.
`list_backends`	Возвращает допустимые строки, которые можно передать в `torch.compile(..., backend="name")`.
`disable`	Эта функция предоставляет как декоратор, так и контекстный менеджер для отключения компиляции функции. Она также предоставляет возможность рекурсивного отключения вызываемых функций.

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/torch.compiler_api.html