diff options
| author |  Gunhan Gulsoy <gunan@google.com> | 2018-09-13 12:31:47 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-13 12:35:54 -0700 |
| commit | d860915b0198ddb96f93e9e97a789af156544dc6 (patch) | |
| tree | d03e340b690e1ec2efa881c21f341fb43bed14ab /tensorflow/contrib/nccl | |
| parent | 54cac449527a6668d5410b6403c1c54d71a9ba82 (diff) | |
Move nccl_rewrite.cc back to tf_kernel_library.
PiperOrigin-RevId: 212858590
Diffstat (limited to 'tensorflow/contrib/nccl')
| -rw-r--r-- | tensorflow/contrib/nccl/BUILD | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/nccl/BUILD b/tensorflow/contrib/nccl/BUILD index 225025e995..9a9d480260 100644 --- a/tensorflow/contrib/nccl/BUILD +++ b/tensorflow/contrib/nccl/BUILD @@ -25,7 +25,7 @@ tf_custom_op_library( name = "python/ops/_nccl_ops.so", srcs = [ "ops/nccl_ops.cc", - ] + if_cuda(["kernels/nccl_rewrite.cc"]), + ], gpu_srcs = if_not_windows_cuda([ "kernels/nccl_manager.cc", "kernels/nccl_manager.h", @@ -74,6 +74,7 @@ tf_kernel_library( "kernels/nccl_manager.cc", "kernels/nccl_manager.h", "kernels/nccl_ops.cc", + "kernels/nccl_rewrite.cc", ]), deps = if_cuda([ "@local_config_nccl//:nccl", |