# Description:
|
# CUDA-platform specific StreamExecutor support code.
|
|
licenses(["notice"]) # Apache 2.0
|
|
load("//tensorflow:tensorflow.bzl", "tf_cc_test")
|
load(
|
"//tensorflow/stream_executor:build_defs.bzl",
|
"stream_executor_friends",
|
"tf_additional_cuda_driver_deps",
|
"tf_additional_cuda_platform_deps",
|
"tf_additional_cudnn_plugin_deps",
|
)
|
load("//tensorflow:tensorflow.bzl", "tf_copts")
|
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda_is_configured")
|
load("//tensorflow/core:platform/default/build_config_root.bzl", "if_static")
|
|
package_group(
|
name = "friends",
|
packages = stream_executor_friends(),
|
)
|
|
package(
|
default_visibility = [":friends"],
|
)
|
|
# Filegroup used to collect source files for the dependency check.
|
filegroup(
|
name = "c_srcs",
|
data = glob([
|
"**/*.cc",
|
"**/*.h",
|
]),
|
)
|
|
cc_library(
|
name = "cuda_platform_id",
|
srcs = ["cuda_platform_id.cc"],
|
hdrs = ["cuda_platform_id.h"],
|
deps = ["//tensorflow/stream_executor:platform"],
|
)
|
|
cc_library(
|
name = "cuda_platform",
|
srcs = if_cuda_is_configured(["cuda_platform.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_platform.h"]),
|
visibility = ["//visibility:public"],
|
deps = if_cuda_is_configured(
|
[
|
":cuda_driver",
|
":cuda_gpu_executor",
|
":cuda_platform_id",
|
"//tensorflow/stream_executor", # buildcleaner: keep
|
"//tensorflow/stream_executor:executor_cache",
|
"//tensorflow/stream_executor:multi_platform_manager",
|
"//tensorflow/stream_executor:stream_executor_pimpl_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
],
|
) + tf_additional_cuda_platform_deps(),
|
alwayslink = True, # Registers itself with the MultiPlatformManager.
|
)
|
|
cc_library(
|
name = "cuda_diagnostics",
|
srcs = if_cuda_is_configured(["cuda_diagnostics.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_diagnostics.h"]),
|
deps = if_cuda_is_configured([
|
"@com_google_absl//absl/container:inlined_vector",
|
"@com_google_absl//absl/strings",
|
"//tensorflow/stream_executor/gpu:gpu_diagnostics_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
]),
|
)
|
|
cc_library(
|
name = "cuda_stub",
|
srcs = if_cuda_is_configured(["cuda_stub.cc"]),
|
textual_hdrs = ["cuda_10_0.inc"],
|
deps = if_cuda_is_configured([
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
]),
|
)
|
|
cc_library(
|
name = "cuda_driver",
|
srcs = if_cuda_is_configured(["cuda_driver.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_driver.h"]),
|
deps = if_cuda_is_configured([
|
":cuda_diagnostics",
|
"@com_google_absl//absl/debugging:leak_check",
|
"@com_google_absl//absl/base",
|
"@com_google_absl//absl/container:inlined_vector",
|
"@com_google_absl//absl/strings",
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor:device_options",
|
"//tensorflow/stream_executor/gpu:gpu_driver_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
] + tf_additional_cuda_driver_deps()) + select({
|
# include dynamic loading implementation only when if_cuda_is_configured and build dynamically
|
"//tensorflow:using_cuda_nvcc_with_dynamic_build": ["cudart_stub"],
|
"//tensorflow:using_cuda_clang_with_dynamic_build": ["cudart_stub"],
|
"//conditions:default": ["//tensorflow/core:cuda"],
|
}),
|
)
|
|
cc_library(
|
name = "cudart_stub",
|
srcs = select({
|
# include dynamic loading implementation only when if_cuda_is_configured and build dynamically
|
"//tensorflow:using_cuda_nvcc_with_dynamic_build": ["cudart_stub.cc"],
|
"//tensorflow:using_cuda_clang_with_dynamic_build": ["cudart_stub.cc"],
|
"//conditions:default": [],
|
}),
|
visibility = ["//visibility:public"],
|
deps = select({
|
"//tensorflow:using_cuda_nvcc_with_dynamic_build": [
|
":cuda_stub",
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
],
|
"//tensorflow:using_cuda_clang_with_dynamic_build": [
|
":cuda_stub",
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
],
|
"//conditions:default": [],
|
}),
|
)
|
|
# The activation library is tightly coupled to the executor library.
|
# TODO(leary) split up cuda_gpu_executor.cc so that this can stand alone.
|
cc_library(
|
name = "cuda_activation_header",
|
hdrs = ["cuda_activation.h"],
|
visibility = ["//visibility:public"],
|
deps = [
|
"//tensorflow/stream_executor/gpu:gpu_activation_header",
|
"//tensorflow/stream_executor/platform",
|
],
|
)
|
|
cc_library(
|
name = "cuda_activation",
|
srcs = [],
|
hdrs = if_cuda_is_configured(["cuda_activation.h"]),
|
deps = if_cuda_is_configured([
|
":cuda_driver",
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor",
|
"//tensorflow/stream_executor:stream_executor_internal",
|
"//tensorflow/stream_executor/gpu:gpu_activation",
|
"//tensorflow/stream_executor/platform",
|
]),
|
)
|
|
cc_library(
|
name = "cuda_gpu_executor_header",
|
textual_hdrs = if_cuda_is_configured(["cuda_gpu_executor.h"]),
|
visibility = ["//visibility:public"],
|
deps = if_cuda_is_configured([
|
":cuda_kernel",
|
"//tensorflow/stream_executor:event",
|
"//tensorflow/stream_executor/gpu:gpu_executor_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
]),
|
)
|
|
cc_library(
|
name = "cublas_stub",
|
srcs = if_cuda_is_configured(["cublas_stub.cc"]),
|
textual_hdrs = glob(["cublas_*.inc"]),
|
deps = if_cuda_is_configured([
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
]),
|
)
|
|
cc_library(
|
name = "cublas_plugin",
|
srcs = if_cuda_is_configured(["cuda_blas.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_blas.h"]),
|
visibility = ["//visibility:public"],
|
deps = if_cuda_is_configured([
|
":cuda_activation",
|
":cuda_gpu_executor",
|
":cuda_platform_id",
|
":cuda_stream",
|
":cuda_timer",
|
":cuda_helpers",
|
"@com_google_absl//absl/strings",
|
"//third_party/eigen3",
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/core:lib_internal",
|
"//tensorflow/stream_executor",
|
"//tensorflow/stream_executor:event",
|
"//tensorflow/stream_executor:host_or_device_scalar",
|
"//tensorflow/stream_executor:plugin_registry",
|
"//tensorflow/stream_executor:scratch_allocator",
|
"//tensorflow/stream_executor:timer",
|
"//tensorflow/stream_executor/gpu:gpu_helpers_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
] + if_static(
|
["@local_config_cuda//cuda:cublas"],
|
[":cublas_stub"],
|
)),
|
alwayslink = True,
|
)
|
|
cc_library(
|
name = "cufft_stub",
|
srcs = if_cuda_is_configured(["cufft_stub.cc"]),
|
textual_hdrs = ["cufft_10_0.inc"],
|
deps = if_cuda_is_configured([
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
]),
|
)
|
|
cc_library(
|
name = "cufft_plugin",
|
srcs = if_cuda_is_configured(["cuda_fft.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_fft.h"]),
|
visibility = ["//visibility:public"],
|
deps = if_cuda_is_configured([
|
":cuda_activation_header",
|
":cuda_gpu_executor_header",
|
":cuda_platform_id",
|
":cuda_stream",
|
":cuda_helpers",
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor:event",
|
"//tensorflow/stream_executor:fft",
|
"//tensorflow/stream_executor:plugin_registry",
|
"//tensorflow/stream_executor:scratch_allocator",
|
"//tensorflow/stream_executor/gpu:gpu_helpers_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
] + if_static(
|
["@local_config_cuda//cuda:cufft"],
|
[":cufft_stub"],
|
)),
|
alwayslink = True,
|
)
|
|
cc_library(
|
name = "cudnn_stub",
|
srcs = if_cuda_is_configured(["cudnn_stub.cc"]),
|
textual_hdrs = glob(["cudnn_*.inc"]),
|
deps = if_cuda_is_configured([
|
"@local_config_cuda//cuda:cudnn_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
]),
|
)
|
|
cc_library(
|
name = "cudnn_plugin",
|
srcs = if_cuda_is_configured(["cuda_dnn.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_dnn.h"]),
|
visibility = ["//visibility:public"],
|
deps = if_cuda_is_configured([
|
":cuda_activation",
|
":cuda_diagnostics",
|
":cuda_driver",
|
":cuda_gpu_executor",
|
":cuda_platform_id",
|
":cuda_stream",
|
":cuda_timer",
|
":cudnn_version",
|
"@com_google_absl//absl/strings",
|
"//third_party/eigen3",
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/core:lib",
|
"//tensorflow/core:lib_internal",
|
"//tensorflow/stream_executor:dnn",
|
"//tensorflow/stream_executor:event",
|
"//tensorflow/stream_executor:plugin_registry",
|
"//tensorflow/stream_executor:scratch_allocator",
|
"//tensorflow/stream_executor:stream_executor_pimpl_header",
|
"//tensorflow/stream_executor:temporary_device_memory",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
]) + tf_additional_cudnn_plugin_deps() + if_cuda_is_configured(if_static(
|
["@local_config_cuda//cuda:cudnn"],
|
[":cudnn_stub"],
|
)),
|
alwayslink = True,
|
)
|
|
cc_library(
|
name = "curand_stub",
|
srcs = if_cuda_is_configured(["curand_stub.cc"]),
|
textual_hdrs = ["curand_10_0.inc"],
|
deps = if_cuda_is_configured([
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
]),
|
)
|
|
cc_library(
|
name = "curand_plugin",
|
srcs = if_cuda_is_configured(["cuda_rng.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_rng.h"]),
|
deps = if_cuda_is_configured([
|
":cuda_activation",
|
":cuda_gpu_executor",
|
":cuda_platform_id",
|
":cuda_stream",
|
":cuda_helpers",
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor:event",
|
"//tensorflow/stream_executor:plugin_registry",
|
"//tensorflow/stream_executor:rng",
|
"//tensorflow/stream_executor/gpu:gpu_helpers_header",
|
"//tensorflow/stream_executor/gpu:gpu_rng_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
] + if_static(
|
["@local_config_cuda//cuda:curand"],
|
[":curand_stub"],
|
)),
|
alwayslink = True,
|
)
|
|
cc_library(
|
name = "cuda_kernel",
|
srcs = if_cuda_is_configured(["cuda_kernel.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_kernel.h"]),
|
deps = if_cuda_is_configured([
|
":cuda_driver",
|
"@local_config_cuda//cuda:cuda_headers",
|
"//tensorflow/stream_executor:event",
|
"//tensorflow/stream_executor:stream_executor_pimpl_header",
|
"//tensorflow/stream_executor/gpu:gpu_kernel_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
]),
|
)
|
|
# TODO(leary) we likely need to canonicalize/eliminate this.
|
cc_library(
|
name = "cuda_helpers",
|
textual_hdrs = if_cuda_is_configured(["cuda_helpers.h"]),
|
deps = if_cuda_is_configured([
|
"//tensorflow/stream_executor/gpu:gpu_helpers_header",
|
]),
|
)
|
|
cc_library(
|
name = "cuda_event",
|
srcs = if_cuda_is_configured(["cuda_event.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_event.h"]),
|
deps = if_cuda_is_configured([
|
":cuda_driver",
|
":cuda_gpu_executor_header",
|
":cuda_stream",
|
"//tensorflow/stream_executor:stream_executor_headers",
|
"//tensorflow/stream_executor/gpu:gpu_event",
|
"//tensorflow/stream_executor/gpu:gpu_stream_header",
|
"//tensorflow/stream_executor/lib",
|
]),
|
)
|
|
cc_library(
|
name = "cuda_stream",
|
srcs = [],
|
hdrs = if_cuda_is_configured(["cuda_stream.h"]),
|
deps = if_cuda_is_configured([
|
":cuda_driver",
|
":cuda_gpu_executor_header",
|
"//tensorflow/stream_executor:stream_executor_headers",
|
"//tensorflow/stream_executor:stream_header",
|
"//tensorflow/stream_executor/gpu:gpu_stream",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
]),
|
)
|
|
cc_library(
|
name = "cuda_timer",
|
srcs = [],
|
hdrs = if_cuda_is_configured(["cuda_timer.h"]),
|
deps = if_cuda_is_configured([
|
":cuda_driver",
|
":cuda_gpu_executor_header",
|
":cuda_stream",
|
"//tensorflow/stream_executor:stream_executor_headers",
|
"//tensorflow/stream_executor/gpu:gpu_timer",
|
"//tensorflow/stream_executor/lib",
|
]),
|
)
|
|
cc_library(
|
name = "cuda_gpu_executor",
|
srcs = if_cuda_is_configured(["cuda_gpu_executor.cc"]),
|
hdrs = if_cuda_is_configured(["cuda_gpu_executor.h"]),
|
deps = if_cuda_is_configured([
|
":cuda_activation",
|
":cuda_diagnostics",
|
":cuda_driver",
|
":cuda_event",
|
":cuda_kernel",
|
":cuda_platform_id",
|
":cuda_stream",
|
":cuda_timer",
|
"@com_google_absl//absl/strings",
|
"//tensorflow/stream_executor:event",
|
"//tensorflow/stream_executor:plugin_registry",
|
"//tensorflow/stream_executor:stream_executor_internal",
|
"//tensorflow/stream_executor:stream_executor_pimpl_header",
|
"//tensorflow/stream_executor:timer",
|
"//tensorflow/stream_executor/gpu:gpu_executor_header",
|
"//tensorflow/stream_executor/lib",
|
"//tensorflow/stream_executor/platform",
|
"//tensorflow/stream_executor/platform:dso_loader",
|
]),
|
alwayslink = True,
|
)
|
|
cc_library(
|
name = "cudnn_version",
|
srcs = ["cudnn_version.cc"],
|
hdrs = ["cudnn_version.h"],
|
deps = [
|
"//tensorflow/core:lib",
|
],
|
)
|
|
tf_cc_test(
|
name = "cudnn_version_test",
|
srcs = ["cudnn_version_test.cc"],
|
deps = [
|
":cudnn_version",
|
"//tensorflow/core:test",
|
"//tensorflow/core:test_main",
|
],
|
)
|
|
cc_library(
|
name = "all_runtime",
|
copts = tf_copts(),
|
visibility = ["//visibility:public"],
|
deps = [
|
":cublas_plugin",
|
":cuda_driver",
|
":cuda_platform",
|
":cudnn_plugin",
|
":cufft_plugin",
|
":curand_plugin",
|
],
|
alwayslink = 1,
|
)
|
