switch aten_headers to use strip_include_prefix instead of includes

Michael Andreas Dagitses · Michael Andreas Dagitses · commit b9c3fd577e9d · 2022-05-20T09:46:16.000-07:00
Pull Request resolved: #77865 This is Bazel best practices and easier to maintain. ghstack-source-id: 156812099 Differential Revision: [D36521515](https://our.internmc.facebook.com/intern/diff/D36521515/)
diff --git a/BUILD.bazel b/BUILD.bazel
@@ -238,15 +238,28 @@ cc_library(
         ":aten_src_ATen_config",
         ":gen_aten",
     ],
-    includes = [
-        "aten/src",
-    ],
+    strip_include_prefix = "aten/src/",
     deps = [
         ":aten_core_headers",
+        ":torch_base_headers",
         "//c10:headers",
     ],
 )
 
+# Temporary library to enable us to use strip_include_prefix =
+# "aten/src/" in aten_headers above. Only use this in aten_headers.
+cc_library(
+    name = "torch_base_headers",
+    hdrs = [
+        "torch/csrc/Export.h",
+        "torch/csrc/jit/frontend/function_schema_parser.h",
+    ],
+    deps = [
+        "//c10/macros",
+        "//c10/util:base",
+    ],
+)
+
 ATEN_COPTS = COMMON_COPTS + [
     "-DCAFFE2_BUILD_MAIN_LIBS",
     "-DHAVE_AVX_CPU_DEFINITION",
