torch.compiler.set_default_backend

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

Задаёт бэкенд по умолчанию для torch.compile, если аргумент backend не указан.

Передача None сбрасывает значение по умолчанию обратно на "inductor".

Параметры:

backend (str | Callable[[...], Any] | None) – Имя бэкенда (строка), вызываемый объект бэкенда или None.

Пример:

>>> torch.compiler.set_default_backend("eager")
>>> torch.compiler.get_default_backend()
'eager'
>>> torch.compiler.set_default_backend(None)  # reset
>>> torch.compiler.get_default_backend()
'inductor'