move generated-autograd-headers to the shared build structure

Michael Andreas Dagitses · Michael Andreas Dagitses · commit 06263e2a62ad · 2022-04-25T05:22:40.000-07:00
Pull Request resolved: #76183 This is a relatively simple target but we have to fix our header expansion to understand generated files. Next step will be to use this in Bazel. ghstack-source-id: 154678047 Differential Revision: [D35820541](https://our.internmc.facebook.com/intern/diff/D35820541/) **NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D35820541/)!
diff --git a/build.bzl b/build.bzl
@@ -42,6 +42,13 @@ def define_targets(rules):
               "--gen_lazy_ts_backend",
     )
 
+    rules.cc_library(
+        name = "generated-autograd-headers",
+        hdrs = [":{}".format(h) for h in _GENERATED_AUTOGRAD_CPP_HEADERS + _GENERATED_AUTOGRAD_PYTHON_HEADERS],
+        tags = ["no-caffe2-headers"],
+        visibility = ["//visibility:public"],
+    )
+
     rules.genrule(
         name = "version_h",
         srcs = [
@@ -74,8 +81,6 @@ _GENERATED_AUTOGRAD_CPP_HEADERS = [
     "torch/csrc/autograd/generated/variable_factories.h",
 ]
 
-GENERATED_AUTOGRAD_H = _GENERATED_AUTOGRAD_CPP_HEADERS + _GENERATED_AUTOGRAD_PYTHON_HEADERS
-
 GENERATED_TESTING_PY = [
     "torch/testing/_internal/generated/annotated_fn_args.py",
 ]
