Skip to content

Commit

Permalink
Update gen.py aoti_fm install dir
Browse files Browse the repository at this point in the history
Summary: make it consistent with all the other install dir

Test Plan: Sandcastle

Differential Revision: D56660301
  • Loading branch information
hl475 authored and facebook-github-bot committed Apr 27, 2024
1 parent 368f521 commit cd6327e
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion torchgen/gen.py
Expand Up @@ -2830,14 +2830,16 @@ def main() -> None:
pathlib.Path(core_install_dir).mkdir(parents=True, exist_ok=True)
ops_install_dir = f"{options.install_dir}/ops"
pathlib.Path(ops_install_dir).mkdir(parents=True, exist_ok=True)
aoti_install_dir = f"{options.install_dir}/aoti_torch/generated"
pathlib.Path(aoti_install_dir).mkdir(parents=True, exist_ok=True)

core_fm = make_file_manager(options=options, install_dir=core_install_dir)
cpu_fm = make_file_manager(options=options)
cpu_vec_fm = make_file_manager(options=options)
cuda_fm = make_file_manager(options=options)
ops_fm = make_file_manager(options=options, install_dir=ops_install_dir)
aoti_fm = make_file_manager(
options=options, install_dir="torch/csrc/inductor/aoti_torch/generated"
options=options, install_dir=aoti_install_dir
)

# Only a limited set of dispatch keys get CPUFunctions.h headers generated
Expand Down

0 comments on commit cd6327e

Please sign in to comment.