torch.compiler.set_default_backend

torch.compiler.set_default_backend(backend)

Set the default backend for torch.compile when no backend argument is specified.

Passing None resets the default back to "inductor".

Parameters:

backend (str | Callable[[...], Any] | None) – A backend name (string), a callable backend, or None.

Example:

>>> 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'