Change dep list of tf_gpu_kernel_library rule.
Change: 117361209
This commit is contained in:
parent
ca7eac9d8d
commit
ca0909cdfc
@ -199,8 +199,8 @@ def tf_gpu_kernel_library(srcs, copts=[], cuda_copts=[], deps=[], hdrs=[],
|
||||
hdrs = hdrs,
|
||||
copts = copts + if_cuda(cuda_copts),
|
||||
deps = deps + if_cuda([
|
||||
"//tensorflow/core:stream_executor",
|
||||
"//tensorflow/core:cuda",
|
||||
"//tensorflow/core:gpu_lib",
|
||||
]),
|
||||
alwayslink=1,
|
||||
**kwargs)
|
||||
|
Loading…
x
Reference in New Issue
Block a user