"//tensorflow/core/kernels:gpu_util_hdrs",
"//tensorflow/core/kernels:ops_util_hdrs",
"//third_party/eigen3",
- "@local_config_cuda//cuda:cudnn_header",
],
alwayslink = 1,
)
"//tensorflow/core/kernels:conv_ops_gpu_hdrs",
"//tensorflow/core/kernels:gpu_util_hdrs",
"//tensorflow/core/kernels:ops_util_hdrs",
- "@local_config_cuda//cuda:cudnn_header",
],
)
name = "utils",
srcs = ["utils.cc"],
hdrs = ["utils.h"],
- cuda_deps = [
- "@local_config_cuda//cuda:cudnn_header",
- ],
visibility = ["//visibility:public"],
deps = [
"//third_party/eigen3",
name = "utils",
srcs = ["utils.cc"],
hdrs = ["utils.h"],
- cuda_deps = [
- "@local_config_cuda//cuda:cudnn_header",
- ],
visibility = ["//visibility:public"],
deps = [
"//third_party/eigen3",
"//tensorflow/core:nn_ops_op_lib",
] + if_cuda([
"@cub_archive//:cub",
- "@local_config_cuda//cuda:cudnn_header",
+ "@local_config_cuda//cuda:cudnn",
]),
)
"//tensorflow/core:lib",
"//tensorflow/core:nn_ops_op_lib",
] + if_cuda([
- "@local_config_cuda//cuda:cudnn_header",
+ "@local_config_cuda//cuda:cudnn",
]),
)
)
cc_library(
- name = "cudnn_header",
- includes = [
- ".",
- "cuda/include",
- ],
- visibility = ["//visibility:public"],
-)
-
-cc_library(
name = "cufft",
srcs = ["cuda/lib/%{cufft_lib}"],
data = ["cuda/lib/%{cufft_lib}"],