switch Bazel to the shared generate-code genrule

Michael Andreas Dagitses · Michael Andreas Dagitses · commit 1a80e27405b5 · 2022-04-15T07:10:37.000-07:00
Pull Request resolved: #75790 We were building it before, but now we use it in downstream rules. This enables us to eliminate the handwritten genrule. ghstack-source-id: 153891038 Differential Revision: [D35645390](https://our.internmc.facebook.com/intern/diff/D35645390/) **NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D35645390/)!
diff --git a/BUILD.bazel b/BUILD.bazel
@@ -146,32 +146,15 @@ libtorch_python_generated_sources = [
         "torch/csrc/autograd/generated/python_return_types.cpp",
 ]
 
-genrule(
-    name = "all_generated_code",
-    srcs = [
-        "aten/src/ATen/native/native_functions.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",
-        "aten/src/ATen/templates/DispatchKeyNativeFunctions.cpp",
-        "aten/src/ATen/templates/DispatchKeyNativeFunctions.h",
-        "aten/src/ATen/templates/RegisterDispatchKey.cpp",
-        "aten/src/ATen/templates/LazyIr.h",
-    ],
-    outs = libtorch_cpp_generated_sources + libtorch_python_generated_sources,
-    cmd = "$(location //tools/setup_helpers:generate_code) --install_dir `dirname $(location torch/csrc/autograd/generated/variable_factories.h)`/../.. --native-functions-path $(location aten/src/ATen/native/native_functions.yaml) --gen_lazy_ts_backend",
-    tools = ["//tools/setup_helpers:generate_code"],
-)
-
 filegroup(
     name = "cpp_generated_code",
-    data = [":all_generated_code"],
+    data = [":generate-code"],
     srcs = libtorch_cpp_generated_sources,
 )
 
 filegroup(
     name = "python_generated_code",
-    data = [":all_generated_code"],
+    data = [":generate-code"],
     srcs = libtorch_python_generated_sources,
 )
 
diff --git a/build.bzl b/build.bzl
@@ -92,4 +92,7 @@ _GENERATED_CPP = [
     "torch/csrc/autograd/generated/python_torch_functions_1.cpp",
     "torch/csrc/autograd/generated/python_torch_functions_2.cpp",
     "torch/csrc/autograd/generated/python_variable_methods.cpp",
+    "torch/csrc/lazy/generated/LazyNativeFunctions.cpp",
+    "torch/csrc/lazy/generated/RegisterAutogradLazy.cpp",
+    "torch/csrc/lazy/generated/RegisterLazy.cpp",
 ]

Original file line number	Diff line number	Diff line change
`@@ -92,4 +92,7 @@ _GENERATED_CPP = [`
`92`	`92`	`"torch/csrc/autograd/generated/python_torch_functions_1.cpp",`
`93`	`93`	`"torch/csrc/autograd/generated/python_torch_functions_2.cpp",`
`94`	`94`	`"torch/csrc/autograd/generated/python_variable_methods.cpp",`
	`95`	`+ "torch/csrc/lazy/generated/LazyNativeFunctions.cpp",`
	`96`	`+ "torch/csrc/lazy/generated/RegisterAutogradLazy.cpp",`
	`97`	`+ "torch/csrc/lazy/generated/RegisterLazy.cpp",`
`95`	`98`	`]`