[tensorflow/stream_executor] Clean up of stream_executor build and misc.
PiperOrigin-RevId: 468583146
diff --git a/tensorflow/stream_executor/BUILD b/tensorflow/stream_executor/BUILD
index 8bb8420..ce5f17d 100644
--- a/tensorflow/stream_executor/BUILD
+++ b/tensorflow/stream_executor/BUILD
@@ -375,10 +375,7 @@
cc_library(
name = "stream_executor_impl",
- deps = [
- "//tensorflow/compiler/xla/stream_executor:dnn_proto_cc_impl",
- "//tensorflow/compiler/xla/stream_executor:stream_executor_impl",
- ],
+ deps = ["//tensorflow/compiler/xla/stream_executor:stream_executor_impl"],
)
cc_library(
diff --git a/tensorflow/stream_executor/cuda/BUILD b/tensorflow/stream_executor/cuda/BUILD
index 304f95c..7ff99c7 100644
--- a/tensorflow/stream_executor/cuda/BUILD
+++ b/tensorflow/stream_executor/cuda/BUILD
@@ -126,7 +126,7 @@
name = "cublas_lib",
actual = select({
"//tensorflow:oss": ":cublas_stub",
- "//conditions:default": "@local_config_cuda//cuda:cublas",
+ "//conditions:default": "//third_party/gpus/cuda:cublas_static",
}),
visibility = ["//visibility:public"],
)
@@ -154,7 +154,7 @@
name = "cublas_lt_lib",
actual = select({
"//tensorflow:oss": ":cublas_lt_stub",
- "//conditions:default": "@local_config_cuda//cuda:cublasLt",
+ "//conditions:default": "//third_party/gpus/cuda:cublas_lt_static",
}),
visibility = ["//visibility:public"],
)
@@ -191,7 +191,7 @@
name = "cufft_lib",
actual = select({
"//tensorflow:oss": ":cufft_stub",
- "//conditions:default": "@local_config_cuda//cuda:cufft",
+ "//conditions:default": "//third_party/gpus/cuda:cufft_static",
}),
visibility = ["//visibility:public"],
)
@@ -217,7 +217,7 @@
name = "cudnn_lib",
actual = select({
"//tensorflow:oss": ":cudnn_stub",
- "//conditions:default": "@local_config_cuda//cuda:cudnn",
+ "//conditions:default": "//third_party/gpus/cudnn:cudnn",
}),
visibility = ["//visibility:public"],
)
@@ -252,7 +252,7 @@
name = "curand_lib",
actual = select({
"//tensorflow:oss": ":curand_stub",
- "//conditions:default": "@local_config_cuda//cuda:curand",
+ "//conditions:default": "//third_party/gpus/cuda:curand_static",
}),
visibility = ["//visibility:public"],
)
@@ -268,7 +268,6 @@
cc_library(
name = "cupti_stub",
- data = if_cuda_is_configured(["@local_config_cuda//cuda:cupti_dsos"]),
deps = [
"//tensorflow/compiler/xla/stream_executor/cuda:cupti_stub",
],
@@ -285,7 +284,7 @@
name = "cusolver_lib",
actual = select({
"//tensorflow:oss": ":cusolver_stub",
- "//conditions:default": "@local_config_cuda//cuda:cusolver",
+ "//conditions:default": "//third_party/gpus/cuda:cusolver_static",
}),
visibility = ["//visibility:public"],
)
@@ -301,7 +300,7 @@
name = "cusparse_lib",
actual = select({
"//tensorflow:oss": ":cusparse_stub",
- "//conditions:default": "@local_config_cuda//cuda:cusparse",
+ "//conditions:default": "//third_party/gpus/cuda:cusparse_static",
}),
visibility = ["//visibility:public"],
)
diff --git a/tensorflow/stream_executor/tpu/BUILD b/tensorflow/stream_executor/tpu/BUILD
index e95949a..a5882bb 100644
--- a/tensorflow/stream_executor/tpu/BUILD
+++ b/tensorflow/stream_executor/tpu/BUILD
@@ -5,6 +5,9 @@
package(
default_visibility = [
"//learning/brain/experimental/dtensor:__subpackages__",
+ "//learning/brain/google/xla/kernels:__subpackages__",
+ "//learning/brain/tfrc/executor:__subpackages__",
+ "//learning/brain/tfrc/runtime/tpu_driver:__subpackages__",
"//tensorflow/compiler/jit:__subpackages__",
"//tensorflow/compiler/mlir:__subpackages__",
"//tensorflow/compiler/xla:__subpackages__",
