OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 00:19:58 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Move tsl/BUILD, tsl.bzl, and tsl.default.bzl to XLA

Browse Source 
PiperOrigin-RevId: 623215553
This commit is contained in:
David Dunleavy 2024-04-09 10:44:27 -07:00 committed by TensorFlower Gardener
parent 8cd7b4166a
commit 068cfff65d
149 changed files with 426 additions and 408 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/BUILD
View File

@ -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",
],

2
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")

2
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")

2
tensorflow/core/platform/build_config.default.bzl
View File

@ -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(

2
tensorflow/core/platform/build_config_root.default.bzl
View File

@ -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": [],

2
tensorflow/core/platform/cloud/BUILD
View File

@ -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")

2
tensorflow/tensorflow.bzl
View File

@ -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",

4
tensorflow/tools/lib_package/BUILD
View File

@ -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": [],
@ -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": [],

2
tensorflow/tools/pip_package/BUILD
View File

@ -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": [],

26
third_party/curl.BUILD vendored
View File

@ -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": [
],
}),
@ -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",
@ -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",
],
@ -405,10 +405,10 @@ 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",
@ -416,8 +416,8 @@ cc_library(
"-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",
@ -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",
],
@ -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",

4
third_party/gloo/gloo.BUILD vendored
View File

@ -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",
],

10
third_party/hwloc/hwloc.BUILD vendored
View File

@ -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",
@ -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",

6
third_party/llvm_openmp/BUILD vendored
View File

@ -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",
@ -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",
})
@ -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,

6
third_party/llvm_openmp/openmp.bzl vendored
View File

@ -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,
})

10
third_party/mkl_dnn/mkldnn_v1.BUILD vendored
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"])
@ -102,7 +102,7 @@ expand_template(
)
_COPTS_LIST = select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": ["-fexceptions"],
}) + [
"-UUSE_MKL",
@ -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,

4
third_party/nccl/build_defs.bzl.tpl vendored
View File

@ -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": [],
}),
)

2
third_party/png.BUILD vendored
View File

@ -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",
],

4
third_party/snappy.BUILD vendored
View File

@ -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",
@ -28,7 +28,7 @@ cc_library(
],
}),
defines = select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": ["HAVE_SYS_UIO_H"],
}),
)

6
third_party/sqlite.BUILD vendored
View File

@ -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",
@ -46,7 +46,7 @@ cc_library(
"SQLITE_OMIT_DEPRECATED",
],
linkopts = select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": [
"-ldl",
"-lpthread",

8
third_party/vulkan_headers/vulkan_headers.BUILD vendored
View File

@ -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"],
@ -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"],

4
third_party/xla/third_party/gloo/gloo.BUILD vendored
View File

@ -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",
],

6
third_party/xla/third_party/llvm_openmp/BUILD vendored
View File

@ -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",
@ -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",
})
@ -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,

6
third_party/xla/third_party/llvm_openmp/openmp.bzl vendored
View File

@ -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,
})

26
third_party/xla/third_party/tsl/third_party/curl.BUILD vendored
View File

@ -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": [
],
}),
@ -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",
@ -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",
],
@ -405,10 +405,10 @@ 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",
@ -416,8 +416,8 @@ cc_library(
"-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",
@ -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",
],
@ -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",

10
third_party/xla/third_party/tsl/third_party/hwloc/hwloc.BUILD vendored
View File

@ -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",
@ -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",

6
third_party/xla/third_party/tsl/third_party/llvm_openmp/BUILD vendored
View File

@ -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",
@ -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",
})
@ -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,

6
third_party/xla/third_party/tsl/third_party/llvm_openmp/openmp.bzl vendored
View File

@ -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,
})

10
third_party/xla/third_party/tsl/third_party/mkl_dnn/mkldnn_v1.BUILD vendored
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"])
@ -102,7 +102,7 @@ expand_template(
)
_COPTS_LIST = select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": ["-fexceptions"],
}) + [
"-UUSE_MKL",
@ -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,

4
third_party/xla/third_party/tsl/third_party/nccl/build_defs.bzl.tpl vendored
View File

@ -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": [],
}),
)

4
third_party/xla/third_party/tsl/third_party/snappy.BUILD vendored
View File

@ -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",
@ -28,7 +28,7 @@ cc_library(
],
}),
defines = select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": ["HAVE_SYS_UIO_H"],
}),
)

2
third_party/xla/third_party/tsl/third_party/zlib.BUILD vendored
View File

@ -33,7 +33,7 @@ cc_library(
],
hdrs = ["zlib.h"],
copts = select({
"@local_tsl//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": [
"-Wno-shift-negative-value",
"-DZ_HAVE_UNISTD_H",

2
third_party/xla/third_party/tsl/tsl/concurrency/BUILD vendored
View File

@ -1,5 +1,5 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
package(

4
third_party/xla/third_party/tsl/tsl/distributed_runtime/BUILD vendored
View File

@ -6,12 +6,12 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = internal_visibility([
"//tsl:internal",
"@local_xla//xla/tsl:internal",
]),
licenses = ["notice"],
)

4
third_party/xla/third_party/tsl/tsl/distributed_runtime/coordination/BUILD vendored
View File

@ -1,11 +1,11 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "if_oss", "internal_visibility", "tsl_gpu_library")
load("@local_xla//xla/tsl:tsl.bzl", "if_oss", "internal_visibility", "tsl_gpu_library")
load("//tsl/platform:build_config.bzl", "tf_proto_library", "tsl_cc_test")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = internal_visibility([
"//tsl:internal",
"@local_xla//xla/tsl:internal",
]),
licenses = ["notice"],
)

6
third_party/xla/third_party/tsl/tsl/distributed_runtime/preemption/BUILD vendored
View File

@ -1,12 +1,12 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "get_compatible_with_portable", "tsl_grpc_cc_dependencies")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "get_compatible_with_portable", "tsl_grpc_cc_dependencies")
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = internal_visibility([
"//tsl:internal",
"@local_xla//xla/tsl:internal",
]),
licenses = ["notice"],
)

6
third_party/xla/third_party/tsl/tsl/distributed_runtime/rpc/BUILD vendored
View File

@ -2,14 +2,14 @@
# RPC communication interfaces and implementations for TensorFlow.
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
load("//tsl/platform:build_config.bzl", "tf_proto_library", "tsl_cc_test")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = internal_visibility([
"//tsl:internal",
"@local_xla//xla/tsl:internal",
]),
licenses = ["notice"],
)

6
third_party/xla/third_party/tsl/tsl/distributed_runtime/rpc/coordination/BUILD vendored
View File

@ -1,11 +1,11 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = internal_visibility([
"//tsl:internal",
"@local_xla//xla/tsl:internal",
]),
licenses = ["notice"],
)

4
third_party/xla/third_party/tsl/tsl/framework/BUILD vendored
View File

@ -8,8 +8,8 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable")
load(
"//tsl/platform:build_config.bzl",
"tsl_cc_test",

32
third_party/xla/third_party/tsl/tsl/framework/contraction/BUILD vendored
View File

@ -1,7 +1,7 @@
load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
load("@bazel_skylib//rules:common_settings.bzl", "bool_flag")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
@ -88,15 +88,15 @@ cc_library(
hdrs = ["eigen_contraction_kernel.h"],
defines = select({
":disable_onednn_contraction_kernel_config": [],
"//tsl:android_x86": [],
"//tsl:arm_any": [
"@local_xla//xla/tsl:android_x86": [],
"@local_xla//xla/tsl:arm_any": [
"TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL",
],
"//tsl:fuchsia_x86_64": [],
"//tsl:ios": [],
"//tsl:linux_ppc64le": [],
"//tsl:linux_s390x": [],
"//tsl:macos_arm64": [],
"@local_xla//xla/tsl:fuchsia_x86_64": [],
"@local_xla//xla/tsl:ios": [],
"@local_xla//xla/tsl:linux_ppc64le": [],
"@local_xla//xla/tsl:linux_s390x": [],
"@local_xla//xla/tsl:macos_arm64": [],
"//conditions:default": [
"TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL",
"TENSORFLOW_USE_MKLDNN_CONTRACTION_KERNEL",
@ -109,13 +109,13 @@ cc_library(
"@eigen_archive//:eigen3",
] + select({
":disable_onednn_contraction_kernel_config": [],
"//tsl:android_x86": [],
"//tsl:arm_any": [],
"//tsl:fuchsia_x86_64": [],
"//tsl:ios": [],
"//tsl:linux_ppc64le": [],
"//tsl:linux_s390x": [],
"//tsl:macos_arm64": [],
"@local_xla//xla/tsl:android_x86": [],
"@local_xla//xla/tsl:arm_any": [],
"@local_xla//xla/tsl:fuchsia_x86_64": [],
"@local_xla//xla/tsl:ios": [],
"@local_xla//xla/tsl:linux_ppc64le": [],
"@local_xla//xla/tsl:linux_s390x": [],
"@local_xla//xla/tsl:macos_arm64": [],
"//conditions:default": ["@onednn//:mkl_dnn"],
}),
)
@ -137,7 +137,7 @@ cc_library(
# Somehow the following code works with fixedpoint, but not here.
# visibility = [
# "//tensorflow:__subpackages__",
# "//tsl:internal",
# "@local_xla//xla/tsl:internal",
# ],
deps = [
"//tsl/framework/fixedpoint",

4
third_party/xla/third_party/tsl/tsl/framework/convolution/BUILD vendored
View File

@ -1,5 +1,5 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
load(
"//tsl/platform:build_config.bzl",
"tsl_cc_test",
@ -64,7 +64,7 @@ filegroup(
# Somehow the following code works with fixedpoint, but not here.
# visibility = [
# "//tensorflow:__subpackages__",
# "//tsl:internal",
# "@local_xla//xla/tsl:internal",
# ],
)

6
third_party/xla/third_party/tsl/tsl/framework/fixedpoint/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
@ -50,7 +50,7 @@ filegroup(
compatible_with = get_compatible_with_portable(),
visibility = internal_visibility([
"//tensorflow:__subpackages__",
"//tsl:internal",
"@local_xla//xla/tsl:internal",
]),
)

4
third_party/xla/third_party/tsl/tsl/lib/core/BUILD vendored
View File

@ -8,8 +8,8 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
# TODO(rdzhabarov): Tighten visibility after migration is complete.

4
third_party/xla/third_party/tsl/tsl/lib/gtl/BUILD vendored
View File

@ -2,8 +2,8 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
load(
"//tsl/platform:build_config.bzl",
"tsl_cc_test",

4
third_party/xla/third_party/tsl/tsl/lib/gtl/subtle/BUILD vendored
View File

@ -1,8 +1,8 @@
# Description:
# gtl subtle packages.
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

4
third_party/xla/third_party/tsl/tsl/lib/hash/BUILD vendored
View File

@ -3,12 +3,12 @@ load(
"cc_library",
)
load(
"//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"if_linux_x86_64",
"internal_visibility",
"tsl_copts",
)
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
load(
"//tsl/platform:build_config.bzl",
"tsl_cc_test",

4
third_party/xla/third_party/tsl/tsl/lib/histogram/BUILD vendored
View File

@ -2,8 +2,8 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
load(
"//tsl/platform:build_config.bzl",
"tsl_cc_test",

4
third_party/xla/third_party/tsl/tsl/lib/io/BUILD vendored
View File

@ -2,8 +2,8 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
package(

2
third_party/xla/third_party/tsl/tsl/lib/io/snappy/BUILD vendored
View File

@ -1,4 +1,4 @@
load("//tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load(
"//tsl/platform:build_config.bzl",
"tsl_cc_test",

4
third_party/xla/third_party/tsl/tsl/lib/math/BUILD vendored
View File

@ -1,5 +1,5 @@
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
load(
"//tsl/platform:build_config.bzl",
"tsl_cc_test",

4
third_party/xla/third_party/tsl/tsl/lib/monitoring/BUILD vendored
View File

@ -2,8 +2,8 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

4
third_party/xla/third_party/tsl/tsl/lib/random/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable")
load(
"//tsl/platform:build_config.bzl",
"tsl_cc_test",

4
third_party/xla/third_party/tsl/tsl/lib/strings/BUILD vendored
View File

@ -2,8 +2,8 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
# copybara:uncomment package(default_applicable_licenses = ["//tensorflow:license"])

8
third_party/xla/third_party/tsl/tsl/platform/BUILD vendored
View File

@ -13,12 +13,12 @@ load(
"cc_library",
)
load(
"//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"if_not_fuchsia",
"internal_visibility",
"tsl_copts",
)
load("//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_xla//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
load(
"//tsl/platform:build_config.bzl",
"tf_cuda_libdevice_path_deps",
@ -578,7 +578,7 @@ filegroup(
"threadpool_interface.h",
"tracing.h",
] + select({
"//tsl:fuchsia": tf_google_mobile_srcs_no_runtime(),
"@local_xla//xla/tsl:fuchsia": tf_google_mobile_srcs_no_runtime(),
"//conditions:default": [
"file_system_helper.cc",
"tracing.cc",
@ -1296,7 +1296,7 @@ cc_library(
srcs = ["test_main.cc"],
copts = tsl_copts(),
linkopts = select({
"//tsl:windows": [],
"@local_xla//xla/tsl:windows": [],
"//conditions:default": ["-lm"],
}),
deps = [

2
third_party/xla/third_party/tsl/tsl/platform/cloud/BUILD vendored
View File

@ -3,7 +3,7 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"if_windows",
"internal_visibility",
"tsl_copts",

2
third_party/xla/third_party/tsl/tsl/platform/cloud/testdata/BUILD vendored
View File

@ -1,6 +1,6 @@
# Cloud test data files.
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

10
third_party/xla/third_party/tsl/tsl/platform/default/BUILD vendored
View File

@ -2,13 +2,13 @@
load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"if_not_fuchsia",
"if_not_windows",
"internal_visibility",
"tsl_copts",
)
load("//tsl:tsl.default.bzl", "filegroup", "tsl_grpc_cc_dependencies")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup", "tsl_grpc_cc_dependencies")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
@ -76,7 +76,7 @@ cc_library(
name = "dso_loader",
srcs = ["dso_loader.cc"] + select({
# include dynamic loading checker only for open source build
"//tsl:oss": ["dlopen_checker.cc"],
"@local_xla//xla/tsl:oss": ["dlopen_checker.cc"],
"//conditions:default": ["dlopen_checker_stub.cc"],
}),
hdrs = ["dso_loader.h"],
@ -341,7 +341,7 @@ cc_library(
copts = tsl_copts(),
defines = ["TF_USE_SNAPPY"] + select({
# TF Additional NUMA defines
"//tsl:with_numa_support": ["TENSORFLOW_USE_NUMA"],
"@local_xla//xla/tsl:with_numa_support": ["TENSORFLOW_USE_NUMA"],
"//conditions:default": [],
}),
tags = [
@ -361,7 +361,7 @@ cc_library(
"@snappy",
] + select({
# TF Additional NUMA dependencies
"//tsl:with_numa_support": [
"@local_xla//xla/tsl:with_numa_support": [
# Don't merge in a single line
"@hwloc",
],

12
third_party/xla/third_party/tsl/tsl/platform/default/build_config.bzl vendored
View File

@ -3,7 +3,7 @@
load("@com_github_grpc_grpc//bazel:generate_cc.bzl", "generate_cc")
load("@com_google_protobuf//:protobuf.bzl", "proto_gen")
load(
"//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"clean_dep",
"if_not_windows",
"if_tsl_link_protobuf",
@ -665,7 +665,7 @@ def tf_additional_lib_hdrs():
clean_dep("//tsl/platform/default:tracing_impl.h"),
clean_dep("//tsl/platform/default:unbounded_work_queue.h"),
] + select({
clean_dep("//tsl:windows"): [
clean_dep("@local_xla//xla/tsl:windows"): [
clean_dep("//tsl/platform/windows:intrinsics_port.h"),
clean_dep("//tsl/platform/windows:stacktrace.h"),
clean_dep("//tsl/platform/windows:subprocess.h"),
@ -720,9 +720,9 @@ def tf_additional_lib_deps():
def tf_additional_core_deps():
return select({
clean_dep("//tsl:android"): [],
clean_dep("//tsl:ios"): [],
clean_dep("//tsl:linux_s390x"): [],
clean_dep("@local_xla//xla/tsl:android"): [],
clean_dep("@local_xla//xla/tsl:ios"): [],
clean_dep("@local_xla//xla/tsl:linux_s390x"): [],
"//conditions:default": [
clean_dep("//tsl/platform/cloud:gcs_file_system"),
],
@ -807,7 +807,7 @@ def tf_protobuf_compiler_deps():
def tf_windows_aware_platform_deps(name):
return select({
clean_dep("//tsl:windows"): [
clean_dep("@local_xla//xla/tsl:windows"): [
clean_dep("//tsl/platform/windows:" + name),
],
"//conditions:default": [

22
third_party/xla/third_party/tsl/tsl/platform/default/build_config_root.bzl vendored
View File

@ -46,53 +46,53 @@ def tf_additional_tpu_ops_deps():
# used for all framework_shared_object platforms including MacOS.
def if_static(extra_deps, otherwise = [], macos = []):
ret = {
str(Label("//tsl:framework_shared_object")): otherwise,
str(Label("@local_xla//xla/tsl:framework_shared_object")): otherwise,
"//conditions:default": extra_deps,
}
if macos:
ret[str(Label("//tsl:macos_with_framework_shared_object"))] = macos
ret[str(Label("@local_xla//xla/tsl:macos_with_framework_shared_object"))] = macos
return select(ret)
def if_static_and_not_mobile(extra_deps, otherwise = []):
return select({
str(Label("//tsl:framework_shared_object")): otherwise,
str(Label("//tsl:android")): otherwise,
str(Label("//tsl:ios")): otherwise,
str(Label("@local_xla//xla/tsl:framework_shared_object")): otherwise,
str(Label("@local_xla//xla/tsl:android")): otherwise,
str(Label("@local_xla//xla/tsl:ios")): otherwise,
"//conditions:default": extra_deps,
})
def if_llvm_aarch32_available(then, otherwise = []):
return select({
str(Label("//tsl:aarch32_or_cross")): then,
str(Label("@local_xla//xla/tsl:aarch32_or_cross")): then,
"//conditions:default": otherwise,
})
def if_llvm_aarch64_available(then, otherwise = []):
return select({
str(Label("//tsl:aarch64_or_cross")): then,
str(Label("@local_xla//xla/tsl:aarch64_or_cross")): then,
"//conditions:default": otherwise,
})
def if_llvm_arm_available(then, otherwise = []):
return select({
str(Label("//tsl:arm_or_cross")): then,
str(Label("@local_xla//xla/tsl:arm_or_cross")): then,
"//conditions:default": otherwise,
})
def if_llvm_powerpc_available(then, otherwise = []):
return select({
str(Label("//tsl:ppc64le_or_cross")): then,
str(Label("@local_xla//xla/tsl:ppc64le_or_cross")): then,
"//conditions:default": otherwise,
})
def if_llvm_system_z_available(then, otherwise = []):
return select({
str(Label("//tsl:s390x_or_cross")): then,
str(Label("@local_xla//xla/tsl:s390x_or_cross")): then,
"//conditions:default": otherwise,
})
def if_llvm_x86_available(then, otherwise = []):
return select({
str(Label("//tsl:x86_or_cross")): then,
str(Label("@local_xla//xla/tsl:x86_or_cross")): then,
"//conditions:default": otherwise,
})

4
third_party/xla/third_party/tsl/tsl/platform/profile_utils/BUILD vendored
View File

@ -5,8 +5,8 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility", "tsl_copts")
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility", "tsl_copts")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

4
third_party/xla/third_party/tsl/tsl/platform/windows/BUILD vendored
View File

@ -5,11 +5,11 @@ load(
# Tensorflow windows-specific implementations of tensorflow/core/platform libraries.
load(
"//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"internal_visibility",
"tsl_copts",
)
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/third_party/tsl/tsl/profiler/backends/cpu/BUILD vendored
View File

@ -1,5 +1,5 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
load("//tsl/platform:build_config_root.bzl", "if_static")
load("//tsl/profiler/builds:build_config.bzl", "tf_profiler_copts")

2
third_party/xla/third_party/tsl/tsl/profiler/builds/BUILD vendored
View File

@ -1,4 +1,4 @@
load("//tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/third_party/tsl/tsl/profiler/builds/build_config.bzl vendored
View File

@ -1,6 +1,6 @@
"""Provides a redirection point for platform specific implementations of Starlark utilities."""
load("//tsl:tsl.bzl", "clean_dep")
load("@local_xla//xla/tsl:tsl.bzl", "clean_dep")
load(
"//tsl/profiler/builds/oss:build_config.bzl",
_tf_profiler_alias = "tf_profiler_alias",

2
third_party/xla/third_party/tsl/tsl/profiler/builds/oss/build_config.bzl vendored
View File

@ -3,7 +3,7 @@
TF profiler build macros for use in OSS.
"""
load("//tsl:tsl.bzl", "cc_header_only_library")
load("@local_xla//xla/tsl:tsl.bzl", "cc_header_only_library")
def tf_profiler_alias(target_dir, name):
return target_dir + "oss:" + name

2
third_party/xla/third_party/tsl/tsl/profiler/convert/BUILD vendored
View File

@ -2,7 +2,7 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load("//tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
load("//tsl/profiler/builds:build_config.bzl", "tf_profiler_copts")

8
third_party/xla/third_party/tsl/tsl/profiler/lib/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "if_not_android", "internal_visibility", "nvtx_headers")
load("//tsl:tsl.default.bzl", "filegroup")
load("@local_xla//xla/tsl:tsl.bzl", "if_not_android", "internal_visibility", "nvtx_headers")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup")
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
load("//tsl/platform:build_config_root.bzl", "if_static")
load(
@ -126,7 +126,7 @@ cc_library(
hdrs = ["profiler_interface.h"],
copts = tf_profiler_copts(),
visibility = internal_visibility([
"//tsl:internal",
"@local_xla//xla/tsl:internal",
"//tsl/profiler:internal",
"//tsl/profiler:xla_profiler_backends",
]),
@ -168,7 +168,7 @@ tsl_cc_test(
cc_library(
name = "profiler_session",
hdrs = ["profiler_session.h"],
visibility = internal_visibility(["//tsl:internal"]),
visibility = internal_visibility(["@local_xla//xla/tsl:internal"]),
deps = [
"//tsl/platform",
"//tsl/platform:errors",

2
third_party/xla/third_party/tsl/tsl/profiler/protobuf/BUILD vendored
View File

@ -1,6 +1,6 @@
# copybara:uncomment(oss-unused) load("//net/grpc/go/build_defs:go_grpc_library.bzl", "go_grpc_library")
# Placeholder: load py_proto_library
load("//tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("//tsl/platform:build_config.bzl", "tf_proto_library")
# copybara:uncomment package(default_applicable_licenses = ["//tensorflow:license"])

4
third_party/xla/third_party/tsl/tsl/profiler/rpc/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
load(
"//tsl/profiler/builds:build_config.bzl",
"tf_profiler_copts",

4
third_party/xla/third_party/tsl/tsl/profiler/rpc/client/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "internal_visibility")
load("//tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
load(
"//tsl/platform:build_config.bzl",
"tf_protos_profiler_service",

2
third_party/xla/third_party/tsl/tsl/profiler/utils/BUILD vendored
View File

@ -1,5 +1,5 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//tsl:tsl.bzl", "internal_visibility")
load("@local_xla//xla/tsl:tsl.bzl", "internal_visibility")
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
load("//tsl/platform:build_config_root.bzl", "if_static")
load("//tsl/profiler/builds:build_config.bzl", "tf_profiler_copts")

4
third_party/xla/third_party/tsl/tsl/protobuf/BUILD vendored
View File

@ -1,6 +1,6 @@
# Placeholder: load py_proto_library
load(
"//tsl:tsl.bzl",
"@local_xla//xla/tsl:tsl.bzl",
"if_google",
"internal_visibility",
)
@ -13,7 +13,7 @@ package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = internal_visibility([
"//tensorflow/core:__subpackages__",
"//tsl:internal",
"@local_xla//xla/tsl:internal",
"//tensorflow_models:__subpackages__",
]),
features = if_google(["-parse_headers"]),

6
third_party/xla/xla/BUILD vendored
View File

@ -1,5 +1,3 @@
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable")
load(
"@local_tsl//tsl/platform:build_config.bzl",
"tf_proto_library",
@ -9,6 +7,8 @@ load("//third_party/compute_library:build_defs.bzl", "if_enable_acl")
# Placeholder: load py_proto_library
load("//xla:xla.bzl", "xla_cc_test", "xla_py_proto_library")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
@ -317,7 +317,7 @@ cc_library(
"statusor.h",
],
linkopts = select({
"@local_tsl//tsl:freebsd": ["-lexecinfo"],
"//xla/tsl:freebsd": ["-lexecinfo"],
"//conditions:default": [],
}),
visibility = ["//visibility:public"],

2
third_party/xla/xla/backends/profiler/BUILD vendored
View File

@ -1,5 +1,5 @@
load(
"@local_tsl//tsl:tsl.bzl",
"//xla/tsl:tsl.bzl",
"if_with_tpu_support",
"internal_visibility",
"tsl_gpu_library",

2
third_party/xla/xla/backends/profiler/cpu/BUILD vendored
View File

@ -1,10 +1,10 @@
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("@local_tsl//tsl/profiler/builds:build_config.bzl", "tf_profiler_copts")
load(
"//xla:xla.bzl",
"xla_cc_test",
)
load("//xla/tsl:tsl.bzl", "internal_visibility")
# copybara:uncomment package(default_applicable_licenses = ["//tensorflow:license"])

12
third_party/xla/xla/backends/profiler/gpu/BUILD vendored
View File

@ -1,11 +1,5 @@
load("@local_config_cuda//cuda:build_defs.bzl", "cuda_library", "if_cuda")
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm")
load(
"@local_tsl//tsl:tsl.bzl",
"internal_visibility",
"tsl_copts",
"tsl_gpu_library",
)
load(
"@local_tsl//tsl/platform:build_config.bzl",
"tf_additional_device_tracer_srcs",
@ -23,6 +17,12 @@ load(
"//xla:xla.bzl",
"xla_cc_test",
)
load(
"//xla/tsl:tsl.bzl",
"internal_visibility",
"tsl_copts",
"tsl_gpu_library",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/backends/profiler/tpu/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@local_tsl//tsl:tsl.bzl", "if_with_tpu_support")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("@local_tsl//tsl/profiler/builds:build_config.bzl", "tf_profiler_copts")
load("//xla/tsl:tsl.bzl", "if_with_tpu_support")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/c/BUILD vendored
View File

@ -1,5 +1,5 @@
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.bzl", "internal_visibility")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/client/BUILD vendored
View File

@ -1,9 +1,9 @@
# Description:
# XLA client libraries.
load("@local_tsl//tsl:tsl.default.bzl", "filegroup")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.default.bzl", "filegroup")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

4
third_party/xla/xla/client/lib/BUILD vendored
View File

@ -1,9 +1,9 @@
# Common computation builders for XLA.
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl:tsl.default.bzl", "filegroup")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tests:build_defs.bzl", "generate_backend_suites", "xla_test")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "filegroup")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/ffi/api/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@local_tsl//tsl:tsl.default.bzl", "filegroup")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.default.bzl", "filegroup")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/hlo/experimental/auto_sharding/BUILD vendored
View File

@ -1,9 +1,9 @@
# Automatic sharding annotation
load("@bazel_skylib//rules:build_test.bzl", "build_test")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_libtpu_portable")
load("@local_tsl//tsl/platform:build_config.bzl", "tf_proto_library")
load("//xla:xla.bzl", "auto_sharding_deps", "auto_sharding_solver_deps", "xla_cc_binary", "xla_cc_test")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_libtpu_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/hlo/ir/BUILD vendored
View File

@ -1,8 +1,8 @@
# Description:
# XLA’s HLO Intermediate Representation implementation.
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.bzl", "internal_visibility")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/mlir/backends/cpu/transforms/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

4
third_party/xla/xla/mlir/framework/ir/BUILD vendored
View File

@ -1,7 +1,7 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

4
third_party/xla/xla/mlir/framework/transforms/BUILD vendored
View File

@ -1,7 +1,7 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library")
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/mlir/memref/transforms/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/mlir/runtime/ir/BUILD vendored
View File

@ -1,6 +1,6 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/mlir/runtime/transforms/BUILD vendored
View File

@ -1,5 +1,4 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load(
"@local_tsl//tsl/platform:build_config_root.bzl",
"if_llvm_aarch64_available",
@ -10,6 +9,7 @@ load(
)
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/mlir/runtime/utils/BUILD vendored
View File

@ -1,5 +1,5 @@
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

4
third_party/xla/xla/mlir/utils/BUILD vendored
View File

@ -1,7 +1,7 @@
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

6
third_party/xla/xla/mlir/xla_cpu/ir/BUILD vendored
View File

@ -1,10 +1,10 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load(
"@local_tsl//tsl:tsl.default.bzl",
"//xla/tsl:tsl.default.bzl",
"get_compatible_with_portable",
)
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

4
third_party/xla/xla/mlir_hlo/BUILD vendored
View File

@ -1,8 +1,8 @@
load("@bazel_skylib//rules:build_test.bzl", "build_test")
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "gentbl_filegroup", "td_library")
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/pjrt/BUILD vendored
View File

@ -1,4 +1,3 @@
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load(
"@local_tsl//tsl/platform:build_config.bzl",
"tf_proto_library",
@ -7,6 +6,7 @@ load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
# Placeholder: load py_proto_library
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.bzl", "internal_visibility")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/pjrt/cpu/BUILD vendored
View File

@ -1,7 +1,7 @@
load("@local_tsl//tsl:tsl.bzl", "if_oss", "internal_visibility")
load("@local_tsl//tsl/platform:build_config.bzl", "tf_proto_library")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.bzl", "if_oss", "internal_visibility")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/pjrt/distributed/BUILD vendored
View File

@ -1,7 +1,7 @@
load("@local_tsl//tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
load("@local_tsl//tsl/platform:build_config.bzl", "tf_proto_library")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.default.bzl", "tsl_grpc_cc_dependencies")
licenses(["notice"])

2
third_party/xla/xla/pjrt/gpu/BUILD vendored
View File

@ -1,10 +1,10 @@
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm")
load("@local_tsl//tsl:tsl.bzl", "if_google", "internal_visibility")
load("@local_tsl//tsl/platform:build_config.bzl", "tf_proto_library")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/stream_executor:build_defs.bzl", "if_cuda_or_rocm", "if_gpu_is_configured")
load("//xla/tsl:tsl.bzl", "if_google", "internal_visibility")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

20
third_party/xla/xla/python/BUILD vendored
View File

@ -1,13 +1,6 @@
load("@bazel_skylib//rules:common_settings.bzl", "bool_flag")
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm")
load(
"@local_tsl//tsl:tsl.bzl",
"if_cuda_or_rocm",
"if_google",
"internal_visibility",
)
load("@local_tsl//tsl:tsl.default.bzl", "tsl_pybind_extension")
load("@local_tsl//tsl/platform:build_config.bzl", "pyx_library", "tf_proto_library")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
@ -21,6 +14,13 @@ load(
"xla_cc_test",
"xla_py_test_deps",
)
load(
"//xla/tsl:tsl.bzl",
"if_cuda_or_rocm",
"if_google",
"internal_visibility",
)
load("//xla/tsl:tsl.default.bzl", "tsl_pybind_extension")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
@ -1273,8 +1273,8 @@ tsl_pybind_extension(
"@llvm-project//mlir:IR",
] + select({
# gloo transport only builds on linux
"@local_tsl//tsl:macos": [],
"@local_tsl//tsl:windows": [],
"//xla/tsl:macos": [],
"//xla/tsl:windows": [],
"//conditions:default": [
"//third_party/gloo:transport_tcp",
"//xla/pjrt/cpu:gloo_collectives",
@ -1282,7 +1282,7 @@ tsl_pybind_extension(
],
}) + select({
# mpitrampoline does not build on windows
"@local_tsl//tsl:windows": [],
"//xla/tsl:windows": [],
"//conditions:default": [
"//xla/pjrt/cpu:mpi_collectives",
],

4
third_party/xla/xla/python/ifrt/BUILD vendored
View File

@ -1,7 +1,7 @@
load("@local_tsl//tsl:tsl.bzl", "internal_visibility")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("@local_tsl//tsl/platform:build_config.bzl", "tf_proto_library")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package_group(
name = "friends",

2
third_party/xla/xla/python/ifrt/ir/BUILD vendored
View File

@ -1,5 +1,5 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/python/ifrt/ir/transforms/BUILD vendored
View File

@ -1,5 +1,5 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library")
load("@local_tsl//tsl:tsl.default.bzl", "get_compatible_with_portable")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

Some files were not shown because too many files have changed in this diff Show More