thunder.executors.torch_compile.TorchCompileExecutor

class thunder.executors.torch_compile.TorchCompileExecutor(name, required_ops=None, *, mode=None)

Bases: FusionExecutor

Fusion executor using torch.compile as the backend.

Parameters:

• name (Hashable) – unique name of the executor

• required_ops (Optional[set]) – set of syms to fuse

• mode (str, optional) –

• name (Hashable) –

• required_ops (set | None) –

• mode (str | None) –

__init__(name, required_ops=None, *, mode=None)

Parameters:

• name (Hashable) –

• required_ops (Optional[set]) –

• mode (Optional[str]) –

Methods

__init__(name[, required_ops, mode])
can_execute(bsym)	rtype: bool
can_execute_or_fuse(bsym)	rtype: bool
can_fuse(bsym)	rtype: bool
fuse(region, fusion_counter)	rtype: BoundSymbol
fusion_pass(trace)	rtype: TraceCtx
get_execution_transform(bsym)	rtype: Optional[Callable]
get_fuel([amount])	rtype: bool
get_grad_transform(sym)	rtype: Optional[Callable]
register_operator(name, *[, like, meta, ...])	rtype: Symbol
register_supported(sym_or_id[, checker, ...])
register_temporary_operation(name, fn, *, ...)	rtype: BoundSymbol
set_fuel(value)

Attributes

implmap
name
opmap
version