thunder.executors.torch_compile.TorchCompileExecutor

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

Bases: FusionExecutor

Fusion executor using torch.compile as the backend.

Parameters:

__init__(name, required_ops=None, *, mode=None)[source]

Parameters:

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

Was this page helpful?