Add bzl_library targets to XLA and TSL

PiperOrigin-RevId: 631181194
ROCm · May 6, 2024 · 33ceb95 · 33ceb95
1 parent c29a067
commit 33ceb95
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/third_party/tsl/tsl/platform/BUILD b/third_party/tsl/tsl/platform/BUILD
@@ -1707,6 +1707,12 @@ bzl_library(
     deps = tf_platform_alias("rules_cc_bzl"),
 )
 
+bzl_library(
+    name = "build_config_root_bzl",
+    srcs = ["build_config_root.bzl"],
+    deps = tf_platform_alias("build_config_root_bzl"),
+)
+
 cc_library(
     name = "retrying_utils",
     srcs = [

diff --git a/third_party/tsl/tsl/platform/default/BUILD b/third_party/tsl/tsl/platform/default/BUILD
@@ -554,6 +554,12 @@ bzl_library(
     srcs = ["rules_cc.bzl"],
 )
 
+bzl_library(
+    name = "build_config_root_bzl",
+    srcs = ["build_config_root.bzl"],
+    # copybara:uncomment parse_tests = False,
+)
+
 # Export source files needed for mobile builds, which do not use granular targets.
 filegroup(
     name = "additional_mobile_srcs_no_runtime",

diff --git a/xla/BUILD b/xla/BUILD
@@ -1,3 +1,4 @@
+load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
 load(
     "@tsl//tsl/platform:build_config.bzl",
     "tf_proto_library",
@@ -1258,3 +1259,20 @@ alias(
     actual = ":empty",
     visibility = ["//visibility:public"],
 )
+
+bzl_library(
+    name = "lit_bzl",
+    srcs = ["lit.bzl"],
+    deps = ["@bazel_skylib//lib:paths"],
+)
+
+bzl_library(
+    name = "xla_bzl",
+    srcs = ["xla.bzl"],
+    deps = [
+        "//xla/tsl:tsl_bzl",
+        "@local_config_rocm//rocm:build_defs_bzl",
+        "@tsl//tsl/platform:build_config_root_bzl",
+        "@tsl//tsl/platform/default:cuda_build_defs_bzl",
+    ],
+)