torch.compiler.load_compiled_function

torch.compiler.load_compiled_function(file, *, f_globals=None, external_data=None) [исходный код]

Загружает AOT-компилированную функцию из файла.

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

В настоящее время этот API является экспериментальным и может измениться.

Параметры:

• file (IOBase) – Файлоподобный объект, содержащий сериализованную скомпилированную функцию.
• f_globals (dict[str, object] | None) – Необязательная глобальная область видимости, в которой находится скомпилированная функция.
• external_data (dict[str, Any] | None) – Необязательные данные для загрузки в среду выполнения скомпилированной функции. Они должны совпадать с данными AOTCompileResult.external_data, возвращаемыми при вызове save_compiled_function().

Возвращает:

Функцию, скомпилированную с помощью torch, с предварительно загруженной с диска компиляцией.

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

Callable[[…], Any]