Skip to content

Commit

Permalink
rm torch_inductor_options
Browse files Browse the repository at this point in the history
  • Loading branch information
kiya00 committed Dec 11, 2024
1 parent 1336946 commit e6d02a3
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions thunder/dynamo/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,9 +27,7 @@ def __init__(self, **thunder_options):
function.
Keyword arguments:
thunder_options: a dictionary of options to pass to :func:`thunder.jit`. Besides all the arguments to :func:`thunder.jit`,
it accepts ``torch_inductor_options`` which are passed to :func:`torch.compile` if part of the graph
is not supported by thunder.
thunder_options: a dictionary of options to pass to :func:`thunder.jit`.
Example:
>>> import torch
Expand Down Expand Up @@ -60,11 +58,9 @@ def __init__(self, **thunder_options):
# Ref to the documentation of `SubgraphInfo` to know more about the information it contains.
self.subgraph_infos: list[SubgraphInfo] = []

torch_inductor_options = thunder_options.pop("torch_inductor_options", {})

self.thunder_options = thunder_options
self._thunder_jit = partial(jit, **thunder_options)
self._torch_compile = partial(torch.compile, **torch_inductor_options)
self._torch_compile = torch.compile

def __call__(self, gm: torch.fx.GraphModule, sample_args: list[torch.SymInt, torch.Tensor]):
gm = remove_empty_autocast(gm)
Expand Down

0 comments on commit e6d02a3

Please sign in to comment.