diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-04-10 16:17:44 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-10 17:28:17 -0700 |
| commit | 72c023d3967a3218cd3d830ce6e57f7c4d87a18c (patch) | |
| tree | 6401e38c3e9cc64f472f31cc84c6b7b80712e718 /third_party/gpus/cuda/BUILD.tpl | |
| parent | 2f5fde8dd977681110714a0564445a406506282c (diff) | |
Add support for solvers based on the Nvidia cuSolver library.
Implement a GPU version of tf.cholesky as a first example.
Change: 152756373
Diffstat (limited to 'third_party/gpus/cuda/BUILD.tpl')
| -rw-r--r-- | third_party/gpus/cuda/BUILD.tpl | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/third_party/gpus/cuda/BUILD.tpl b/third_party/gpus/cuda/BUILD.tpl index e101f9fbd8..474c972a4c 100644 --- a/third_party/gpus/cuda/BUILD.tpl +++ b/third_party/gpus/cuda/BUILD.tpl @@ -91,6 +91,16 @@ cc_library( ) cc_library( + name = "cusolver", + srcs = ["lib/%{cusolver_lib}"], + data = ["lib/%{cusolver_lib}"], + includes = ["include/"], + linkstatic = 1, + linkopts = ["-lgomp"], + visibility = ["//visibility:public"], +) + +cc_library( name = "cudnn", srcs = ["lib/%{cudnn_lib}"], data = ["lib/%{cudnn_lib}"], |