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