experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix cuda remote build setup.

Browse Source 
PiperOrigin-RevId: 211842211
This commit is contained in:
A. Unique TensorFlower 2018-09-06 11:41:42 -07:00 committed by TensorFlower Gardener
parent c19d517511
commit 0d76eadeb2
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
third_party/gpus/cuda/remote.BUILD.tpl vendored
View File

@ -74,6 +74,11 @@ alias(
actual = "%{remote_cuda_repo}/cuda:cudnn", actual = "%{remote_cuda_repo}/cuda:cudnn",
) )
alias(
name = "cudnn_header",
actual = "%{remote_cuda_repo}/cuda:cudnn_header",
)
alias( alias(
name = "cufft", name = "cufft",
actual = "%{remote_cuda_repo}/cuda:cufft", actual = "%{remote_cuda_repo}/cuda:cufft",