aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-09-06 11:41:42 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-06 11:47:22 -0700
commit0d76eadeb21ee9ff607838e65cb89533086109a6 (patch)
treeeaa7a958df96c67f0bc4a4ce79690579775030f6 /third_party
parentc19d517511e7c3739f626243d19813aec3226184 (diff)
Fix cuda remote build setup.
PiperOrigin-RevId: 211842211
Diffstat (limited to 'third_party')
-rw-r--r--third_party/gpus/cuda/remote.BUILD.tpl5
1 files changed, 5 insertions, 0 deletions
diff --git a/third_party/gpus/cuda/remote.BUILD.tpl b/third_party/gpus/cuda/remote.BUILD.tpl
index f774def5e6..100c7bb7c4 100644
--- a/third_party/gpus/cuda/remote.BUILD.tpl
+++ b/third_party/gpus/cuda/remote.BUILD.tpl
@@ -75,6 +75,11 @@ alias(
)
alias(
+ name = "cudnn_header",
+ actual = "%{remote_cuda_repo}/cuda:cudnn_header",
+)
+
+alias(
name = "cufft",
actual = "%{remote_cuda_repo}/cuda:cufft",
)