| # Description: |
| # CUDA-platform specific StreamExecutor support code. |
| |
| load("//tensorflow:tensorflow.bzl", "check_deps", "tf_cc_test", "tf_copts", "tf_cuda_cc_test") |
| load("//tensorflow:tensorflow.bzl", "filegroup") |
| load( |
| "//tensorflow/stream_executor:build_defs.bzl", |
| "stream_executor_friends", |
| "tf_additional_cuda_driver_deps", |
| "tf_additional_cuda_platform_deps", |
| "tf_additional_cudnn_plugin_copts", |
| "tf_additional_cudnn_plugin_deps", |
| ) |
| load( |
| "//tensorflow/tsl/platform/default:cuda_build_defs.bzl", |
| "if_cuda_is_configured", |
| ) |
| load( |
| "//tensorflow/core/platform:build_config_root.bzl", |
| "if_static", |
| "tf_cuda_tests_tags", |
| ) |
| load( |
| "//tensorflow/core/platform:rules_cc.bzl", |
| "cc_library", |
| ) |
| |
| package( |
| default_visibility = [":friends"], |
| features = ["-layering_check"], |
| licenses = ["notice"], |
| ) |
| |
| package_group( |
| name = "friends", |
| packages = stream_executor_friends(), |
| ) |
| |
| cc_library( |
| name = "cuda_platform_id", |
| srcs = ["cuda_platform_id.cc"], |
| hdrs = ["cuda_platform_id.h"], |
| deps = ["//tensorflow/compiler/xla/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", |
| ":cuda_activation", |
| "//tensorflow/compiler/xla/stream_executor", # buildcleaner: keep |
| "//tensorflow/compiler/xla/stream_executor:executor_cache", |
| "//tensorflow/compiler/xla/stream_executor:multi_platform_manager", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_pimpl_header", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform", |
| ], |
| ) + tf_additional_cuda_platform_deps() + [ |
| "@com_google_absl//absl/base", |
| "@com_google_absl//absl/base:core_headers", |
| "@com_google_absl//absl/memory", |
| "@com_google_absl//absl/strings", |
| "@com_google_absl//absl/strings:str_format", |
| "@com_google_absl//absl/synchronization", |
| ], |
| 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/compiler/xla/stream_executor/gpu:gpu_diagnostics_header", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform", |
| ]) + ["@com_google_absl//absl/strings:str_format"], |
| ) |
| |
| cc_library( |
| name = "cuda_stub", |
| srcs = if_cuda_is_configured(["cuda_stub.cc"]), |
| textual_hdrs = glob(["cuda_*.inc"]), |
| deps = if_cuda_is_configured([ |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| ) |
| |
| # Buildozer can not remove dependencies inside select guards, so we have to use |
| # an intermediate target. |
| cc_library(name = "ptxas_wrapper") |
| |
| # Buildozer can not remove dependencies inside select guards, so we have to use |
| # an intermediate target. |
| cc_library(name = "fatbinary_wrapper") |
| |
| 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:node_hash_map", |
| "@com_google_absl//absl/container:inlined_vector", |
| "@com_google_absl//absl/strings", |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor:device_options", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_driver_header", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform", |
| "//tensorflow/compiler/xla/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:is_cuda_enabled_and_oss": ["cudart_stub"], |
| "//conditions:default": ["//tensorflow/core:cuda"], |
| }) + [ |
| "@com_google_absl//absl/base:core_headers", |
| "@com_google_absl//absl/memory", |
| "@com_google_absl//absl/strings:str_format", |
| "@com_google_absl//absl/synchronization", |
| ], |
| ) |
| |
| tf_cuda_cc_test( |
| name = "stream_search_test", |
| size = "small", |
| srcs = ["stream_search_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| "//tensorflow/compiler/xla/stream_executor", |
| "//tensorflow/compiler/xla/stream_executor/host:host_platform", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cuda_cc_test( |
| name = "cuda_driver_test", |
| srcs = ["cuda_driver_test.cc"], |
| tags = tf_cuda_tests_tags() + [ |
| "no_cuda_asan", # TODO(b/171512140): re-enable. |
| ], |
| deps = [ |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "@local_config_cuda//cuda:cuda_headers", |
| ], |
| ) |
| |
| tf_cuda_cc_test( |
| name = "memcpy_test", |
| srcs = ["memcpy_test.cc"], |
| tags = tf_cuda_tests_tags() + [ |
| "no_cuda_asan", # TODO(b/171512140): re-enable. |
| ], |
| deps = [ |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| cc_library( |
| name = "cudart_stub", |
| srcs = select({ |
| # include dynamic loading implementation only when if_cuda_is_configured and build dynamically |
| "//tensorflow:is_cuda_enabled_and_oss": ["cudart_stub.cc"], |
| "//conditions:default": [], |
| }), |
| textual_hdrs = glob(["cuda_runtime_*.inc"]), |
| visibility = ["//visibility:public"], |
| deps = select({ |
| "//tensorflow:is_cuda_enabled_and_oss": [ |
| ":cuda_stub", |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/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/compiler/xla/stream_executor/gpu:gpu_activation_header", |
| "//tensorflow/compiler/xla/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/compiler/xla/stream_executor", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_internal", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_activation", |
| "//tensorflow/compiler/xla/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/compiler/xla/stream_executor:event", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_executor_header", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/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/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| ) |
| |
| alias( |
| name = "cublas_lib", |
| actual = select({ |
| "//tensorflow:oss": ":cublas_stub", |
| "//conditions:default": "@local_config_cuda//cuda:cublas", |
| }), |
| visibility = ["//visibility:public"], |
| ) |
| |
| cc_library( |
| name = "cublas_lt_header", |
| hdrs = if_cuda_is_configured([ |
| "cuda_blas_lt.h", |
| "cuda_blas_utils.h", |
| ]), |
| visibility = ["//visibility:public"], |
| deps = if_cuda_is_configured([ |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/core/platform:status", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor:host_or_device_scalar", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_headers", |
| "//tensorflow/compiler/xla/stream_executor/platform", |
| ]), |
| ) |
| |
| cc_library( |
| name = "cublas_lt_stub", |
| srcs = if_cuda_is_configured(["cublasLt_stub.cc"]), |
| textual_hdrs = glob(["cublasLt_*.inc"]), |
| deps = if_cuda_is_configured([ |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| ) |
| |
| alias( |
| name = "cublas_lt_lib", |
| actual = select({ |
| "//tensorflow:oss": ":cublas_lt_stub", |
| "//conditions:default": "@local_config_cuda//cuda:cublasLt", |
| }), |
| visibility = ["//visibility:public"], |
| ) |
| |
| cc_library( |
| name = "cublas_plugin", |
| srcs = if_cuda_is_configured([ |
| "cuda_blas.cc", |
| "cuda_blas_lt.cc", |
| ]), |
| hdrs = if_cuda_is_configured([ |
| "cuda_blas.h", |
| "cuda_blas_lt.h", |
| ]), |
| visibility = ["//visibility:public"], |
| deps = if_cuda_is_configured([ |
| ":cublas_lib", |
| ":cublas_lt_lib", |
| ":cuda_activation", |
| ":cuda_blas_utils", |
| ":cuda_gpu_executor", |
| ":cuda_helpers", |
| ":cuda_platform_id", |
| ":cuda_stream", |
| ":cuda_timer", |
| "@com_google_absl//absl/base:core_headers", |
| "@com_google_absl//absl/strings", |
| "@com_google_absl//absl/strings:str_format", |
| "@com_google_absl//absl/synchronization", |
| "@com_google_absl//absl/types:span", |
| "//third_party/eigen3", |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla:status_macros", |
| "//tensorflow/core/platform:tensor_float_32_hdr_lib", |
| "//tensorflow/core/platform:thread_annotations", |
| "//tensorflow/compiler/xla/stream_executor:device_memory", |
| "//tensorflow/compiler/xla/stream_executor:host_or_device_scalar", |
| "//tensorflow/compiler/xla/stream_executor:scratch_allocator", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_headers", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_activation_header", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_executor_header", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_helpers_header", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_stream_header", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_timer_header", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_types_header", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform", |
| ]) + if_static(["//tensorflow/core/platform:tensor_float_32_utils"]), |
| alwayslink = True, |
| ) |
| |
| cc_library( |
| name = "cuda_blas_utils", |
| srcs = if_cuda_is_configured(["cuda_blas_utils.cc"]), |
| hdrs = if_cuda_is_configured(["cuda_blas_utils.h"]), |
| deps = if_cuda_is_configured([ |
| ":cublas_lib", |
| "@com_google_absl//absl/strings", |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_headers", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| ]), |
| ) |
| |
| cc_library( |
| name = "cufft_stub", |
| srcs = if_cuda_is_configured(["cufft_stub.cc"]), |
| textual_hdrs = glob(["cufft_*.inc"]), |
| deps = if_cuda_is_configured([ |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| ) |
| |
| alias( |
| name = "cufft_lib", |
| actual = select({ |
| "//tensorflow:oss": ":cufft_stub", |
| "//conditions:default": "@local_config_cuda//cuda:cufft", |
| }), |
| visibility = ["//visibility:public"], |
| ) |
| |
| 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", |
| ":cufft_lib", |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor:event", |
| "//tensorflow/compiler/xla/stream_executor:fft", |
| "//tensorflow/compiler/xla/stream_executor:plugin_registry", |
| "//tensorflow/compiler/xla/stream_executor:scratch_allocator", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_helpers_header", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| 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/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ":cudnn_version", |
| ]), |
| ) |
| |
| alias( |
| name = "cudnn_lib", |
| actual = select({ |
| "//tensorflow:oss": ":cudnn_stub", |
| "//conditions:default": "@local_config_cuda//cuda:cudnn", |
| }), |
| visibility = ["//visibility:public"], |
| ) |
| |
| cc_library( |
| name = "cuda_dnn_headers", |
| textual_hdrs = ["cuda_dnn.h"], |
| deps = if_cuda_is_configured([ |
| ":cuda_activation_header", |
| "//tensorflow/compiler/xla/stream_executor:dnn", |
| "//tensorflow/compiler/xla/stream_executor:plugin_registry", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| ]), |
| ) |
| |
| cc_library( |
| name = "cudnn_plugin", |
| srcs = if_cuda_is_configured(["cuda_dnn.cc"]), |
| hdrs = if_cuda_is_configured(["cuda_dnn.h"]), |
| copts = tf_additional_cudnn_plugin_copts(), |
| 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", |
| ":cudnn_lib", |
| "@com_google_absl//absl/base:core_headers", |
| "@com_google_absl//absl/memory", |
| "@com_google_absl//absl/strings", |
| "@com_google_absl//absl/types:span", |
| "@cudnn_frontend_archive//:cudnn_frontend", |
| "//third_party/eigen3", |
| "@local_config_cuda//cuda:cuda_headers", |
| "@local_config_cuda//cuda:cudnn_header", |
| "//tensorflow/core:lib_internal", |
| "//tensorflow/core/platform:tensor_float_32_utils", |
| "//tensorflow/core/util:determinism", |
| "//tensorflow/compiler/xla/stream_executor:dnn", |
| "//tensorflow/compiler/xla/stream_executor:event", |
| "//tensorflow/compiler/xla/stream_executor:plugin_registry", |
| "//tensorflow/compiler/xla/stream_executor:scratch_allocator", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_pimpl_header", |
| "//tensorflow/compiler/xla/stream_executor:temporary_device_memory", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform", |
| ]) + tf_additional_cudnn_plugin_deps() + ["@com_google_absl//absl/synchronization"], |
| 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/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| ) |
| |
| alias( |
| name = "curand_lib", |
| actual = select({ |
| "//tensorflow:oss": ":curand_stub", |
| "//conditions:default": "@local_config_cuda//cuda:curand", |
| }), |
| visibility = ["//visibility:public"], |
| ) |
| |
| 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", |
| ":curand_lib", |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor:event", |
| "//tensorflow/compiler/xla/stream_executor:plugin_registry", |
| "//tensorflow/compiler/xla/stream_executor:rng", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_helpers_header", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_rng_header", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| alwayslink = True, |
| ) |
| |
| cc_library( |
| name = "cupti_stub", |
| srcs = if_cuda_is_configured(["cupti_stub.cc"]), |
| data = if_cuda_is_configured(["@local_config_cuda//cuda:cupti_dsos"]), |
| textual_hdrs = ["cupti_10_0.inc"], |
| deps = if_cuda_is_configured([ |
| "@local_config_cuda//cuda:cuda_headers", |
| "@local_config_cuda//cuda:cupti_headers", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| ) |
| |
| cc_library( |
| name = "cusolver_stub", |
| srcs = if_cuda_is_configured(["cusolver_stub.cc"]), |
| textual_hdrs = glob(["cusolver_dense_*.inc"]), |
| deps = if_cuda_is_configured([ |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| ) |
| |
| alias( |
| name = "cusolver_lib", |
| actual = select({ |
| "//tensorflow:oss": ":cusolver_stub", |
| "//conditions:default": "@local_config_cuda//cuda:cusolver", |
| }), |
| visibility = ["//visibility:public"], |
| ) |
| |
| cc_library( |
| name = "cusparse_stub", |
| srcs = if_cuda_is_configured(["cusparse_stub.cc"]), |
| textual_hdrs = glob(["cusparse_*.inc"]), |
| deps = if_cuda_is_configured([ |
| "@local_config_cuda//cuda:cuda_headers", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]), |
| ) |
| |
| alias( |
| name = "cusparse_lib", |
| actual = select({ |
| "//tensorflow:oss": ":cusparse_stub", |
| "//conditions:default": "@local_config_cuda//cuda:cusparse", |
| }), |
| visibility = ["//visibility:public"], |
| ) |
| |
| 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/compiler/xla/stream_executor:event", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_pimpl_header", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_kernel_header", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/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/compiler/xla/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/compiler/xla/stream_executor:stream_executor_headers", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_event", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_stream_header", |
| "//tensorflow/compiler/xla/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/compiler/xla/stream_executor:stream_executor_headers", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_stream", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/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/compiler/xla/stream_executor:stream_executor_headers", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_timer", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| ]), |
| ) |
| |
| cc_library( |
| name = "cuda_asm_compiler", |
| srcs = if_cuda_is_configured(["cuda_asm_compiler.cc"]), |
| deps = if_cuda_is_configured([ |
| "//tensorflow/core:lib_proto_parsing", |
| "//tensorflow/compiler/xla/stream_executor/gpu:asm_compiler", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_driver_header", |
| ]), |
| ) |
| |
| 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", |
| ":cuda_asm_compiler", |
| "@com_google_absl//absl/strings", |
| "//tensorflow/compiler/xla/stream_executor:event", |
| "//tensorflow/compiler/xla/stream_executor:plugin_registry", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_internal", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_pimpl_header", |
| "//tensorflow/compiler/xla/stream_executor:timer", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_executor_header", |
| "//tensorflow/compiler/xla/stream_executor/lib", |
| "//tensorflow/compiler/xla/stream_executor/platform", |
| "//tensorflow/compiler/xla/stream_executor/platform:dso_loader", |
| ]) + [ |
| "@com_google_absl//absl/strings:str_format", |
| ], |
| alwayslink = True, |
| ) |
| |
| cc_library( |
| name = "cudnn_version", |
| srcs = ["cudnn_version.cc"], |
| hdrs = ["cudnn_version.h"], |
| deps = [ |
| "@com_google_absl//absl/strings", |
| ], |
| ) |
| |
| 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, |
| ) |
| |
| # To avoid duplication, check that the C++ or python library does not depend on |
| # the stream executor cuda plugins. Targets that want to use cuda APIs should |
| # instead depend on the dummy plugins in //tensorflow/tsl/platform/default/build_config |
| # and use header only targets. |
| check_deps( |
| name = "cuda_plugins_check_deps", |
| disallowed_deps = if_static( |
| [], |
| otherwise = [ |
| ":all_runtime", |
| ":cuda_driver", |
| ":cuda_platform", |
| ":cudnn_plugin", |
| ":cufft_plugin", |
| ":curand_plugin", |
| "//tensorflow/compiler/xla/stream_executor:cuda_platform", |
| ], |
| ), |
| deps = [ |
| "//tensorflow:tensorflow_cc", |
| "//tensorflow/python:pywrap_tensorflow_internal", |
| ], |
| ) |
| |
| tf_cuda_cc_test( |
| name = "redzone_allocator_test", |
| srcs = ["redzone_allocator_test.cc"], |
| tags = tf_cuda_tests_tags() + [ |
| "no_cuda_asan", # TODO(b/171512140): re-enable. |
| ], |
| deps = [ |
| ":cuda_activation", |
| ":cuda_gpu_executor", |
| "//tensorflow/compiler/xla/stream_executor:device_memory_allocator", |
| "//tensorflow/compiler/xla/stream_executor:event", |
| "//tensorflow/compiler/xla/stream_executor:kernel", |
| "//tensorflow/compiler/xla/stream_executor/gpu:gpu_asm_opts", |
| "//tensorflow/compiler/xla/stream_executor/gpu:redzone_allocator", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "//tensorflow/core/platform:stream_executor_no_cuda", |
| "//tensorflow/tsl/platform/default/build_config:stream_executor_cuda", |
| ], |
| ) |
| |
| filegroup( |
| name = "cudnn_wrappers", |
| srcs = glob(["*.inc"]), |
| visibility = ["//visibility:public"], |
| ) |
