Skip to content
This repository was archived by the owner on Aug 1, 2025. It is now read-only.

Use Low overhead version AOT Module #113

Merged
merged 10 commits into from
Apr 6, 2022
Merged

Use Low overhead version AOT Module #113

Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
53f0b8e
Fix torchbench.py --nothing option
jansel Mar 31, 2022
f7b68d0
Skip tacotron2 in torchbench.py
jansel Mar 31, 2022
3cdfd70
Merge branch 'nothing202203' into tacotron202203
jansel Mar 31, 2022
d72ff46
revert
jansel Mar 31, 2022
ecfa457
Merge branch 'nothing202203' into main
jansel Mar 31, 2022
72e3552
Merge branch 'tacotron202203' into main
jansel Mar 31, 2022
27270b3
Fix --nothing
jansel Mar 31, 2022
26aa43b
Merge branch 'main' of github.com:facebookresearch/torchdynamo into main
jansel Mar 31, 2022
b90d477
Merge branch 'facebookresearch:main' into main
jansel Apr 4, 2022
b77e6e9
Using low overhead version of AOT Module
anijain2305 Apr 6, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions torchdynamo/optimizations/backends.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -511,9 +511,9 @@ def _wrapped_bw_compiler(*args, **kwargs):
bw_compiler = kwargs.get("bw_compiler") or kwargs["fw_compiler"]
kwargs["bw_compiler"] = _wrapped_bw_compiler

from functorch.compile import aot_module
from functorch.compile import aot_module_simplified

return subgraph.wrap_returns(aot_module(subgraph.model, **kwargs))
return aot_module_simplified(subgraph.model, **kwargs)


def tvm_compile(jit_mod, example_inputs, log_file=None, **kwargs):
Expand Down