gen_aten output into aten/src/ATen/

Michael Andreas Dagitses · Michael Andreas Dagitses · commit c7d8b887d1bf · 2022-05-18T12:02:46.000-07:00
Pull Request resolved: #77773 This is more natural for the Bazel build. ghstack-source-id: 156697863 Differential Revision: [D36487559](https://our.internmc.facebook.com/intern/diff/D36487559/) **NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D36487559/)!
diff --git a/build.bzl b/build.bzl
@@ -47,7 +47,7 @@ def define_targets(rules):
 
     gen_aten_cmd = " ".join([
         "$(location //torchgen:gen)",
-        "--install_dir=$(RULEDIR)",
+        "--install_dir=$(RULEDIR)/aten/src/ATen/",
         "--source-path aten/src/ATen",
     ] + (["--static_dispatch_backend CPU"] if rules.is_cpu_static_dispatch_build() else []))
 
@@ -69,15 +69,18 @@ def define_targets(rules):
         name = "gen_aten",
         srcs = gen_aten_srcs,
         tools = ["//torchgen:gen"],
-        outs = gen_aten_outs,
+        outs = ["aten/src/ATen/" + out for out in gen_aten_outs],
         cmd = gen_aten_cmd,
+        # This currently clashes with the existing Bazel generated
+        # files.
+        tags = ["-bazel"],
     )
 
     rules.genrule(
         name = "gen_aten_hip",
         srcs = gen_aten_srcs,
         tools = ["//torchgen:gen"],
-        outs = gen_aten_outs_cuda,
+        outs = ["aten/src/ATen/" + out for out in gen_aten_outs_cuda],
         cmd = gen_aten_cmd + " --rocm",
         features = ["-create_bazel_outputs"],
         tags = ["-bazel"],
