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

Add missing CUDA header dependency to NCCL.

Browse Source 
PiperOrigin-RevId: 251696588
This commit is contained in:
A. Unique TensorFlower 2019-06-05 12:37:22 -07:00 committed by TensorFlower Gardener
parent 5c727ed0f4
commit 3ab0f000e4
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
third_party/nccl/archive.BUILD vendored
View File

@ -25,6 +25,7 @@ cc_library(
name = "include_hdrs",
hdrs = process_srcs(glob(["src/include/*.h"])),
strip_include_prefix = "include",
deps = ["@local_config_cuda//cuda:cuda_headers"],
)
device_srcs = process_srcs([