frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add aot_inductor as a test backend for benchmarking (#105221)

Browse Source 
Summary:
Original PR at https://github.com/pytorch/pytorch/pull/104977. Landing from fbcode instead.

Add an aot_inductor backend (Export+AOTInductor) in the benchmarking harness. Note it is not a dynamo backend.

Moved files from torch/_inductor/aot_inductor_include to torch/csrc/inductor as a more standard way for exposing headers
Created a caching function in benchmarks/dynamo/common.py for compiling, loading and caching the .so file, as a proxy for a pure C++ deployment, but easier for benchmarking.

Differential Revision: D47452591

Pull Request resolved: https://github.com/pytorch/pytorch/pull/105221
Approved by: https://github.com/jansel
This commit is contained in:
Bin Bao
2023-07-18 13:16:36 +00:00
committed by PyTorch MergeBot
parent 671a21926f
commit b10de43c0a
12 changed files with 162 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
setup.py
View File

@ -1203,9 +1203,8 @@ def main():
'include/THH/*.h*',
'include/THH/generic/*.h',
'include/sleef.h',
"_inductor/codegen/*.h",
"_inductor/aot_inductor_include/*.cpp",
"_inductor/aot_inductor_include/*.h",
'_inductor/codegen/*.cpp',
'_inductor/codegen/*.h',
'share/cmake/ATen/*.cmake',
'share/cmake/Caffe2/*.cmake',
'share/cmake/Caffe2/public/*.cmake',