1 licenses(["restricted"]) 2 3 load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda_is_configured") 4 load("//tensorflow/core:platform/default/build_config_root.bzl", "if_static") 5 6 STREAM_EXECUTOR_HEADERS = glob([ 7 "*.h", 8 "cuda/*.h", 9 "host/*.h", 10 "lib/*.h", 11 "lib/gtl/*.h", 12 "platform/**/*.h", 13 ]) 14 15 cc_library( 16 name = "stream_executor_impl", 17 srcs = glob( 18 [ 19 "*.cc", 20 "host/*.cc", 21 "cuda/cuda_platform_id.cc", 22 "lib/*.cc", 23 "platform/default/*.cc", 24 ], 25 exclude = [ 26 "**/*_test.cc", 27 ], 28 ), 29 hdrs = STREAM_EXECUTOR_HEADERS, 30 linkopts = select({ 31 "//tensorflow:freebsd": [], 32 "//conditions:default": ["-ldl"], 33 }), 34 visibility = ["//visibility:public"], 35 deps = [ 36 "//tensorflow/core:lib", 37 "@local_config_cuda//cuda:cuda_headers", 38 ], 39 alwayslink = 1, 40 ) 41 42 cc_library( 43 name = "stream_executor", 44 hdrs = STREAM_EXECUTOR_HEADERS, 45 visibility = ["//visibility:public"], 46 deps = [ 47 "//tensorflow/core:lib", 48 "@local_config_cuda//cuda:cuda_headers", 49 ] + if_static([":stream_executor_impl"]), 50 ) 51 52 cc_library( 53 name = "cuda_platform", 54 srcs = if_cuda_is_configured( 55 glob( 56 [ 57 "cuda/*.cc", 58 ], 59 exclude = ["cuda/cuda_platform_id.cc"], 60 ), 61 ), 62 copts = select({ 63 "//tensorflow:windows": ["/DNOGDI"], 64 "//conditions:default": [], 65 }), 66 linkopts = select({ 67 "//tensorflow:freebsd": [], 68 "//conditions:default": ["-ldl"], 69 }), 70 visibility = ["//visibility:public"], 71 deps = [ 72 ":stream_executor", 73 "//tensorflow/core:lib", 74 "//tensorflow/core/kernels:ops_util", 75 "@local_config_cuda//cuda:cuda_headers", 76 ] + if_cuda_is_configured([ 77 "//tensorflow/core:cuda", 78 "@local_config_cuda//cuda:cublas", 79 "@local_config_cuda//cuda:cuda_driver", 80 "@local_config_cuda//cuda:cudnn", 81 "@local_config_cuda//cuda:cufft", 82 "@local_config_cuda//cuda:curand", 83 ]), 84 alwayslink = 1, 85 ) 86
