blob: 51170e4531d03dac7f9eb3d391ec18d7a8536582 [file] [log] [blame]
licenses(["notice"]) # Apache 2.0
package(default_visibility = ["//tensorflow/stream_executor:__subpackages__"])
load("//tensorflow/core/platform:default/build_config_root.bzl", "if_static")
load("//tensorflow:tensorflow.bzl", "tf_copts")
cc_library(
name = "platform",
textual_hdrs = [
"initialize.h",
],
deps = ["//tensorflow/core:lib"],
)
cc_library(
name = "dso_loader",
srcs = ["dso_loader.cc"] + if_static(
["dlopen_checker_stub.cc"],
["dlopen_checker.cc"],
),
hdrs = ["dso_loader.h"],
copts = tf_copts(),
deps = [
"//tensorflow/stream_executor:platform",
"//tensorflow/stream_executor/lib",
"//tensorflow/stream_executor/platform",
"@com_google_absl//absl/strings",
"@local_config_cuda//cuda:cuda_headers",
"@local_config_tensorrt//:tensorrt_headers",
],
)