From 37fb636b7fe0dde597c33d9d194576bd0a6cf195 Mon Sep 17 00:00:00 2001
From: mikey dagitses <mikeyd@fb.com>
Date: Thu, 5 May 2022 16:31:35 -0700
Subject: [PATCH] fix package violation caused by D35587412 (#76808)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76808

This reached into aten/TARGETS in fbcode.
ghstack-source-id: 155484095

Test Plan: Verified manually.

Reviewed By: dreiss, malfet

Differential Revision: D36128458

fbshipit-source-id: c7447b3a40fe905993e799d211241e72183f8acb
(cherry picked from commit b68eb7a45d8973fadab2dfcafcbb0f63801abd40)
---
 BUILD.bazel | 26 +++++++++++++++++---------
 build.bzl   | 14 +++++++-------
 2 files changed, 24 insertions(+), 16 deletions(-)

diff --git a/BUILD.bazel b/BUILD.bazel
index 79bf1b4018b29..2327db6564f19 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -1906,12 +1906,20 @@ test_suite(
 # as if they are from this package, so we alias them for
 # compatibility.
 
-alias(
-    name = "native_functions.yaml",
-    actual = "aten/src/ATen/native/native_functions.yaml",
-)
-
-alias(
-    name = "tags.yaml",
-    actual = "aten/src/ATen/native/tags.yaml",
-)
+[
+    alias(
+        name = paths.basename(path),
+        actual = path,
+    )
+    for path in [
+        "aten/src/ATen/templates/DispatchKeyNativeFunctions.cpp",
+        "aten/src/ATen/templates/DispatchKeyNativeFunctions.h",
+        "aten/src/ATen/templates/LazyIr.h",
+        "aten/src/ATen/templates/RegisterDispatchKey.cpp",
+        "aten/src/ATen/native/native_functions.yaml",
+        "aten/src/ATen/native/tags.yaml",
+        "aten/src/ATen/native/ts_native_functions.yaml",
+        "torch/csrc/lazy/core/shape_inference.h",
+        "torch/csrc/lazy/ts_backend/ts_native_functions.cpp",
+    ]
+]
diff --git a/build.bzl b/build.bzl
index 6302d55b84404..d4d3a5898f5de 100644
--- a/build.bzl
+++ b/build.bzl
@@ -25,15 +25,15 @@ def define_targets(rules):
     rules.genrule(
         name = "generate-code",
         srcs = [
-            "aten/src/ATen/native/ts_native_functions.yaml",
-            "aten/src/ATen/templates/DispatchKeyNativeFunctions.cpp",
-            "aten/src/ATen/templates/DispatchKeyNativeFunctions.h",
-            "aten/src/ATen/templates/LazyIr.h",
-            "aten/src/ATen/templates/RegisterDispatchKey.cpp",
-            "torch/csrc/lazy/core/shape_inference.h",
-            "torch/csrc/lazy/ts_backend/ts_native_functions.cpp",
+            ":DispatchKeyNativeFunctions.cpp",
+            ":DispatchKeyNativeFunctions.h",
+            ":LazyIr.h",
+            ":RegisterDispatchKey.cpp",
             ":native_functions.yaml",
+            ":shape_inference.h",
             ":tags.yaml",
+            ":ts_native_functions.cpp",
+            ":ts_native_functions.yaml",
         ],
         tools = ["//tools/setup_helpers:generate_code"],
         outs = _GENERATED_CPP + GENERATED_AUTOGRAD_H + GENERATED_LAZY_H + GENERATED_TESTING_PY,