Skip to content

Commit

Permalink
Move tsl/BUILD, tsl.bzl, and tsl.default.bzl to XLA
Browse files Browse the repository at this point in the history 
PiperOrigin-RevId: 623215553
  • Loading branch information
@ddunl @tensorflower-gardener
ddunl authored and tensorflower-gardener committed Apr 9, 2024
1 parent 8cd7b41 commit 068cfff
Show file tree
Hide file tree
Showing 149 changed files with 426 additions and 408 deletions.
2 changes: 1 addition & 1 deletion tensorflow/BUILD
View file
Expand Up @@ -1117,8 +1117,8 @@ bzl_library(
"@local_config_cuda//cuda:build_defs_bzl",
"@local_config_rocm//rocm:build_defs_bzl",
"@local_config_tensorrt//:build_defs_bzl",
"@local_tsl//tsl:tsl_bzl",
"@local_tsl//tsl/platform/default:cuda_build_defs_bzl",
"@local_xla//xla/tsl:tsl_bzl",
"@local_xla//xla/tsl/mkl:build_defs_bzl",
"@rules_java//java:rules",
],
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/compiler/mlir/stablehlo/BUILD
View file
@@ -1,4 +1,4 @@
load("@local_tsl//tsl:tsl.default.bzl", "tsl_pybind_extension")
load("@local_xla//xla/tsl:tsl.default.bzl", "tsl_pybind_extension")
load("//tensorflow:pytype.default.bzl", "pytype_strict_library")
load("//tensorflow:strict.default.bzl", "py_strict_test")

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/core/data/service/snapshot/BUILD
View file
@@ -1,6 +1,6 @@
# Distributed snapshot library.

load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
load("//tensorflow:tensorflow.bzl", "tf_cc_test")
load("//tensorflow:tensorflow.default.bzl", "tf_grpc_cc_dependencies", "tf_kernel_library")
load("//tensorflow/core/platform:build_config.bzl", "tf_protos_profiler_service")
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/core/platform/build_config.default.bzl
View file
Expand Up @@ -2,7 +2,7 @@

load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm")
load(
"@local_tsl//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"if_libtpu",
)
load(
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/core/platform/build_config_root.default.bzl
View file
Expand Up @@ -3,7 +3,7 @@
# unused in TSL
def tf_additional_plugin_deps():
return select({
str(Label("@local_tsl//tsl:with_xla_support")): [
str(Label("@local_xla//xla/tsl:with_xla_support")): [
str(Label("//tensorflow/compiler/jit")),
],
"//conditions:default": [],
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/core/platform/cloud/BUILD
View file
Expand Up @@ -2,7 +2,7 @@
# Cloud file system implementation.

load(
"@local_tsl//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"tsl_copts",
)
load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tensorflow.bzl
View file
Expand Up @@ -69,7 +69,7 @@ load(
"rocm_copts",
)
load(
"@local_tsl//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"tsl_gpu_library",
_cc_header_only_library = "cc_header_only_library",
_if_cuda_or_rocm = "if_cuda_or_rocm",
Expand Down
4 changes: 2 additions & 2 deletions tensorflow/tools/lib_package/BUILD
View file
Expand Up @@ -160,7 +160,7 @@ genrule(
"@com_github_googlecloudplatform_google_cloud_cpp//:LICENSE",
],
}) + select({
"@local_tsl//tsl:with_numa_support": [
"@local_xla//xla/tsl:with_numa_support": [
"@hwloc//:COPYING",
],
"//conditions:default": [],
Expand Down Expand Up @@ -203,7 +203,7 @@ genrule(
"@com_github_googlecloudplatform_google_cloud_cpp//:LICENSE",
],
}) + select({
"@local_tsl//tsl:with_numa_support": [
"@local_xla//xla/tsl:with_numa_support": [
"@hwloc//:COPYING",
],
"//conditions:default": [],
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/pip_package/BUILD
View file
Expand Up @@ -113,7 +113,7 @@ filegroup(
"@com_github_googlecloudplatform_google_cloud_cpp//:LICENSE",
],
}) + select({
"@local_tsl//tsl:with_numa_support": [
"@local_xla//xla/tsl:with_numa_support": [
"@hwloc//:COPYING",
],
"//conditions:default": [],
Expand Down
26 changes: 13 additions & 13 deletions third_party/curl.BUILD
View file
Expand Up @@ -353,13 +353,13 @@ cc_library(
"lib/ws.c",
"lib/ws.h",
] + select({
"@local_tsl//tsl:macos": [
"@local_xla//xla/tsl:macos": [
"lib/vtls/sectransp.c",
],
"@local_tsl//tsl:ios": [
"@local_xla//xla/tsl:ios": [
"lib/vtls/sectransp.c",
],
"@local_tsl//tsl:windows": CURL_WIN_SRCS,
"@local_xla//xla/tsl:windows": CURL_WIN_SRCS,
"//conditions:default": [
],
}),
Expand All @@ -378,7 +378,7 @@ cc_library(
"include/curl/websockets.h",
],
copts = select({
"@local_tsl//tsl:windows": CURL_WIN_COPTS,
"@local_xla//xla/tsl:windows": CURL_WIN_COPTS,
"//conditions:default": [
"-Iexternal/curl/lib",
"-D_GNU_SOURCE",
Expand All @@ -391,10 +391,10 @@ cc_library(
"-Wno-string-plus-int",
],
}) + select({
"@local_tsl//tsl:macos": [
"@local_xla//xla/tsl:macos": [
"-fno-constant-cfstrings",
],
"@local_tsl//tsl:windows": [
"@local_xla//xla/tsl:windows": [
# See curl.h for discussion of write size and Windows
"/DCURL_MAX_WRITE_SIZE=16384",
],
Expand All @@ -405,19 +405,19 @@ cc_library(
defines = ["CURL_STATICLIB"],
includes = ["include"],
linkopts = select({
"@local_tsl//tsl:android": [
"@local_xla//xla/tsl:android": [
"-pie",
],
"@local_tsl//tsl:macos": [
"@local_xla//xla/tsl:macos": [
"-Wl,-framework",
"-Wl,CoreFoundation",
"-Wl,-framework",
"-Wl,SystemConfiguration",
"-Wl,-framework",
"-Wl,Security",
],
"@local_tsl//tsl:ios": [],
"@local_tsl//tsl:windows": [
"@local_xla//xla/tsl:ios": [],
"@local_xla//xla/tsl:windows": [
"-DEFAULTLIB:ws2_32.lib",
"-DEFAULTLIB:advapi32.lib",
"-DEFAULTLIB:crypt32.lib",
Expand All @@ -431,8 +431,8 @@ cc_library(
deps = [
"@zlib",
] + select({
"@local_tsl//tsl:ios": [],
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:ios": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": [
"@boringssl//:ssl",
],
Expand Down Expand Up @@ -538,7 +538,7 @@ cc_binary(
"src/tool_xattr.h",
],
copts = select({
"@local_tsl//tsl:windows": CURL_BIN_WIN_COPTS,
"@local_xla//xla/tsl:windows": CURL_BIN_WIN_COPTS,
"//conditions:default": [
"-Iexternal/curl/lib",
"-D_GNU_SOURCE",
Expand Down
4 changes: 2 additions & 2 deletions third_party/gloo/gloo.BUILD
View file
Expand Up @@ -57,8 +57,8 @@ cc_library(
"gloo/rendezvous/prefix_store.cc",
"gloo/rendezvous/store.cc",
] + select({
"@local_tsl//tsl:macos": [],
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:macos": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": [
"gloo/common/linux.cc",
],
Expand Down
10 changes: 5 additions & 5 deletions third_party/hwloc/hwloc.BUILD
View file
Expand Up @@ -50,7 +50,7 @@ expand_template(
name = "include_hwloc_autogen_config_h",
out = "include/hwloc/autogen/config.h",
substitutions = select({
"@local_tsl//tsl:linux_x86_64": _INCLUDE_HWLOC_AUTOIGEN_CONFIG_H_LINUX_SUBS,
"@local_xla//xla/tsl:linux_x86_64": _INCLUDE_HWLOC_AUTOIGEN_CONFIG_H_LINUX_SUBS,
"//conditions:default": _INCLUDE_HWLOC_AUTOIGEN_CONFIG_H_COMMON_SUBS,
}),
template = "include/hwloc/autogen/config.h.in",
Expand Down Expand Up @@ -259,21 +259,21 @@ cc_library(
"include/private/private.h",
"include/private/xml.h",
] + select({
"@local_tsl//tsl:linux_x86_64": [
"@local_xla//xla/tsl:linux_x86_64": [
"hwloc/topology-linux.c",
"hwloc/topology-x86.c",
"include/hwloc/linux.h",
"include/private/cpuid-x86.h",
],
"@local_tsl//tsl:linux_aarch64": [
"@local_xla//xla/tsl:linux_aarch64": [
"hwloc/topology-linux.c",
"include/hwloc/linux.h",
],
"@local_tsl//tsl:linux_ppc64le": [
"@local_xla//xla/tsl:linux_ppc64le": [
"hwloc/topology-linux.c",
"include/hwloc/linux.h",
],
"@local_tsl//tsl:freebsd": [
"@local_xla//xla/tsl:freebsd": [
"hwloc/topology-freebsd.c",
"hwloc/topology-x86.c",
"include/private/cpuid-x86.h",
Expand Down
6 changes: 3 additions & 3 deletions third_party/llvm_openmp/BUILD
View file
Expand Up @@ -12,7 +12,7 @@ load(
"libiomp5_cc_binary",
)
load(
"@local_tsl//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"if_linux_x86_64",
"if_macos",
"if_windows",
Expand All @@ -36,7 +36,7 @@ py_binary(
)

kmp_i18n_os_type = select({
"@local_tsl//tsl:windows": "win",
"@local_xla//xla/tsl:windows": "win",
"//conditions:default": "lin",
})

Expand Down Expand Up @@ -114,7 +114,7 @@ omp_vars_win = {
}

omp_all_cmake_vars = select({
"@local_tsl//tsl:windows": cmake_var_string(
"@local_xla//xla/tsl:windows": cmake_var_string(
dict_add(
omp_vars,
omp_vars_win,
Expand Down
6 changes: 3 additions & 3 deletions third_party/llvm_openmp/openmp.bzl
View file
Expand Up @@ -33,9 +33,9 @@ def dict_add(*dictionaries):

def select_os_specific(L, M, W):
return select({
"@local_tsl//tsl:linux_x86_64": L,
"@local_tsl//tsl:macos": M,
"@local_tsl//tsl:windows": W,
"@local_xla//xla/tsl:linux_x86_64": L,
"@local_xla//xla/tsl:macos": M,
"@local_xla//xla/tsl:windows": W,
"//conditions:default": L,
})

Expand Down
10 changes: 5 additions & 5 deletions third_party/mkl_dnn/mkldnn_v1.BUILD
View file
@@ -1,6 +1,6 @@
load("@bazel_skylib//rules:expand_template.bzl", "expand_template")
load("@local_tsl//third_party/mkl_dnn:build_defs.bzl", "if_mkldnn_openmp")
load("@local_tsl//tsl:tsl.bzl", "tf_openmp_copts")
load("@local_xla//xla/tsl:tsl.bzl", "tf_openmp_copts")
load("@local_xla//xla/tsl/mkl:build_defs.bzl", "if_mkl", "if_mkl_ml")

exports_files(["LICENSE"])
Expand Down Expand Up @@ -102,7 +102,7 @@ expand_template(
)

_COPTS_LIST = select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": ["-fexceptions"],
}) + [
"-UUSE_MKL",
Expand Down Expand Up @@ -171,9 +171,9 @@ cc_library(
includes = _INCLUDES_LIST,
# TODO(penpornk): Use lrt_if_needed from tensorflow.bzl instead.
linkopts = select({
"@local_tsl//tsl:linux_aarch64": ["-lrt"],
"@local_tsl//tsl:linux_x86_64": ["-lrt"],
"@local_tsl//tsl:linux_ppc64le": ["-lrt"],
"@local_xla//xla/tsl:linux_aarch64": ["-lrt"],
"@local_xla//xla/tsl:linux_x86_64": ["-lrt"],
"@local_xla//xla/tsl:linux_ppc64le": ["-lrt"],
"//conditions:default": [],
}),
textual_hdrs = _TEXTUAL_HDRS_LIST,
Expand Down
4 changes: 2 additions & 2 deletions third_party/nccl/build_defs.bzl.tpl
View file
Expand Up @@ -329,8 +329,8 @@ def cuda_rdc_library(name, hdrs = None, copts = None, linkstatic = True, **kwarg
out = dlink_cc,
gpu_archs = cuda_gpu_architectures(),
nvlink_args = select({
"@local_tsl//tsl:linux_x86_64": ["--cpu-arch=X86_64"],
"@local_tsl//tsl:linux_ppc64le": ["--cpu-arch=PPC64LE"],
"@local_xla//xla/tsl:linux_x86_64": ["--cpu-arch=X86_64"],
"@local_xla//xla/tsl:linux_ppc64le": ["--cpu-arch=PPC64LE"],
"//conditions:default": [],
}),
)
Expand Down
2 changes: 1 addition & 1 deletion third_party/png.BUILD
View file
Expand Up @@ -33,7 +33,7 @@ cc_library(
"intel/filter_sse2_intrinsics.c",
"intel/intel_init.c",
],
"@local_tsl//tsl:linux_ppc64le": [
"@local_xla//xla/tsl:linux_ppc64le": [
"powerpc/filter_vsx_intrinsics.c",
"powerpc/powerpc_init.c",
],
Expand Down
4 changes: 2 additions & 2 deletions third_party/snappy.BUILD
View file
Expand Up @@ -19,7 +19,7 @@ cc_library(
],
hdrs = ["snappy.h"],
copts = ["-DHAVE_CONFIG_H"] + select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": [
"-fno-exceptions",
"-Wno-sign-compare",
Expand All @@ -28,7 +28,7 @@ cc_library(
],
}),
defines = select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": ["HAVE_SYS_UIO_H"],
}),
)
Expand Down
6 changes: 3 additions & 3 deletions third_party/sqlite.BUILD
View file
Expand Up @@ -11,10 +11,10 @@ SQLITE_COPTS = [
"-D_FILE_OFFSET_BITS=64",
"-D_REENTRANT=1",
] + select({
"@local_tsl//tsl:windows": [
"@local_xla//xla/tsl:windows": [
"-DSQLITE_MAX_TRIGGER_DEPTH=100",
],
"@local_tsl//tsl:macos": [
"@local_xla//xla/tsl:macos": [
"-Os",
"-DHAVE_GMTIME_R=1",
"-DHAVE_LOCALTIME_R=1",
Expand Down Expand Up @@ -46,7 +46,7 @@ cc_library(
"SQLITE_OMIT_DEPRECATED",
],
linkopts = select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": [
"-ldl",
"-lpthread",
Expand Down
8 changes: 4 additions & 4 deletions third_party/vulkan_headers/vulkan_headers.BUILD
View file
Expand Up @@ -53,8 +53,8 @@ cc_library(
name = "vulkan_hpp",
srcs =
select({
"@local_tsl//tsl:macos": [],
"@local_tsl//tsl:ios": [],
"@local_xla//xla/tsl:macos": [],
"@local_xla//xla/tsl:ios": [],
"//conditions:default": ["tensorflow/vulkan_hpp_dispatch_loader_dynamic.cc"],
}),
hdrs = ["include/vulkan/vulkan.hpp"],
Expand All @@ -65,8 +65,8 @@ cc_library(
"VULKAN_HPP_TYPESAFE_CONVERSION",
"VULKAN_HPP_TYPESAFE_EXPLICIT",
] + select({
"@local_tsl//tsl:macos": [],
"@local_tsl//tsl:ios": [],
"@local_xla//xla/tsl:macos": [],
"@local_xla//xla/tsl:ios": [],
"//conditions:default": ["VULKAN_HPP_DISPATCH_LOADER_DYNAMIC"],
}),
includes = ["include"],
Expand Down
4 changes: 2 additions & 2 deletions third_party/xla/third_party/gloo/gloo.BUILD
View file
Expand Up @@ -57,8 +57,8 @@ cc_library(
"gloo/rendezvous/prefix_store.cc",
"gloo/rendezvous/store.cc",
] + select({
"@local_tsl//tsl:macos": [],
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:macos": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": [
"gloo/common/linux.cc",
],
Expand Down

0 comments on commit 068cfff

Please sign in to comment.