27 lines
593 B
Plaintext
27 lines
593 B
Plaintext
# Adopted from: https://github.com/NVIDIA/TRTorch/blob/master/third_party/cudnn/local/BUILD
|
|
|
|
cc_library(
|
|
name = "cudnn_headers",
|
|
hdrs = ["include/cudnn.h"] + glob([
|
|
"include/cudnn+.h",
|
|
"include/cudnn_*.h",
|
|
]),
|
|
includes = ["include/"],
|
|
visibility = ["//visibility:private"],
|
|
)
|
|
|
|
cc_import(
|
|
name = "cudnn_lib",
|
|
shared_library = "lib/x86_64-linux-gnu/libcudnn.so",
|
|
visibility = ["//visibility:private"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "cudnn",
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"cudnn_headers",
|
|
"cudnn_lib",
|
|
],
|
|
)
|