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

Keep same autogenerated files structure between fbcode and OSS builds (#35951)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35951

Change generate_code to keep folder structure the same regardless of whether install path is provide
Amend build_variables.bzl accordingly

Another preliminary step to merge https://github.com/pytorch/pytorch/pull/35220

Test Plan: CI

Reviewed By: EscapeZero, seemethere

Differential Revision: D20839410

fbshipit-source-id: 02297560a7e48aa7c6271f7a8517fc4a1ab35271
This commit is contained in:
Nikita Shulga
2020-04-03 12:26:17 -07:00
committed by Facebook GitHub Bot
parent ba3cec867f
commit 7b04772c51
3 changed files with 39 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tools/setup_helpers/generate_code.py
View File

@ -34,27 +34,32 @@ def generate_code(ninja_global=None,
from tools.jit.gen_jit_dispatch import gen_jit_dispatch
# Build ATen based Variable classes
autograd_gen_dir = install_dir or 'torch/csrc/autograd/generated'
jit_gen_dir = install_dir or 'torch/csrc/jit/generated'
install_dir = install_dir or 'torch/csrc'
autograd_gen_dir = os.path.join(install_dir, 'autograd', 'generated')
jit_gen_dir = os.path.join(install_dir, 'jit', 'generated')
for d in (autograd_gen_dir, jit_gen_dir):
if not os.path.exists(d):
os.makedirs(d)
runfiles_dir = os.environ.get("RUNFILES_DIR", None)
data_dir = os.path.join(runfiles_dir, 'pytorch') if runfiles_dir else ''
autograd_dir = os.path.join(data_dir, 'tools', 'autograd')
tools_jit_templates = os.path.join(data_dir, 'tools', 'jit', 'templates')
if subset == "pybindings" or not subset:
gen_autograd_python(declarations_path or DECLARATIONS_PATH, autograd_gen_dir, 'tools/autograd')
gen_autograd_python(declarations_path or DECLARATIONS_PATH, autograd_gen_dir, autograd_dir)
if subset == "libtorch" or not subset:
gen_autograd(
declarations_path or DECLARATIONS_PATH,
autograd_gen_dir,
'tools/autograd',
autograd_dir,
disable_autograd=disable_autograd,
disable_trace=disable_trace,
)
gen_jit_dispatch(
declarations_path or DECLARATIONS_PATH,
jit_gen_dir,
'tools/jit/templates',
tools_jit_templates,
disable_autograd=disable_autograd,
selected_op_list_path=selected_op_list_path)