1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
# Adopted from: https://github.com/tensorflow/runtime/blob/master/third_party/rules_cuda/private/BUILD.local_cuda
# Library targets are created corresponding to BUILD.bazel's needs.
cc_library(
name = "cuda_headers",
hdrs = glob([
"include/**",
"targets/x86_64-linux/include/**",
]),
includes = [
"include",
"targets/x86_64-linux/include",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "cuda_driver",
srcs = ["lib64/stubs/libcuda.so"],
visibility = ["//visibility:public"],
)
cc_library(
name = "cuda",
srcs = ["targets/x86_64-linux/lib/libcudart.so"],
visibility = ["//visibility:public"],
deps = [":cuda_headers"],
)
cc_library(
name = "cufft",
srcs = ["targets/x86_64-linux/lib/libcufft.so"],
visibility = ["//visibility:public"],
)
cc_library(
name = "cublas",
srcs = [
"targets/x86_64-linux/lib/libcublasLt.so",
"targets/x86_64-linux/lib/libcublas.so",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "curand",
srcs = ["targets/x86_64-linux/lib/libcurand.so"],
visibility = ["//visibility:public"],
)
cc_library(
name = "cusolver",
srcs = ["targets/x86_64-linux/lib/libcusolver.so"],
visibility = ["//visibility:public"],
)
cc_library(
name = "cusparse",
srcs = ["targets/x86_64-linux/lib/libcusparse.so"],
visibility = ["//visibility:public"],
)
cc_library(
name = "nvrtc",
srcs = [
"targets/x86_64-linux/lib/libnvrtc.so",
"targets/x86_64-linux/lib/libnvrtc-builtins.so",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "nvToolsExt",
srcs = [ "lib64/libnvToolsExt.so"],
visibility = ["//visibility:public"],
)
|
