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

Reviewed By: frank-wei

Differential Revision: D56660301
  • Loading branch information
hl475 authored and facebook-github-bot committed Apr 29, 2024
1 parent 06b845d commit 25abcd7
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions torchgen/gen.py
Expand Up @@ -2830,15 +2830,14 @@ 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}/torch/csrc/inductor/aoti_torch/generated"

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"
)
aoti_fm = make_file_manager(options=options, install_dir=aoti_install_dir)

# Only a limited set of dispatch keys get CPUFunctions.h headers generated
# for them; this is the set
Expand Down

0 comments on commit 25abcd7

Please sign in to comment.