xref: /aosp_15_r20/external/pytorch/third_party/cudnn.BUILD (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1# Adopted from: https://github.com/NVIDIA/TRTorch/blob/master/third_party/cudnn/local/BUILD
2
3cc_library(
4    name = "cudnn_headers",
5    hdrs = ["include/cudnn.h"] + glob([
6        "include/cudnn+.h",
7        "include/cudnn_*.h",
8    ]),
9    includes = ["include/"],
10    visibility = ["//visibility:private"],
11)
12
13cc_import(
14    name = "cudnn_lib",
15    shared_library = "lib64/libcudnn.so",
16    visibility = ["//visibility:private"],
17)
18
19cc_library(
20    name = "cudnn",
21    visibility = ["//visibility:public"],
22    deps = [
23        "cudnn_headers",
24        "cudnn_lib",
25    ],
26)
27