aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/mpi_collectives
diff options
context:
space:
mode:
authorGravatar Yong Tang <yong.tang.github@outlook.com>2018-06-20 02:12:08 +0000
committerGravatar Yong Tang <yong.tang.github@outlook.com>2018-06-28 15:47:05 +0000
commit1b79572c2d47c1b7d82c670c27b9641335b7c025 (patch)
tree16e8056c8bc383275c05d6a3e1b9f0209fb1f77a /tensorflow/contrib/mpi_collectives
parent51ddb66cdf1444998827e03c4b5f592841ee6255 (diff)
Fix MPI build issue with bazel
While trying to build tensorflow with MPI through: ``` ... $ bazel build -s --verbose_failures --config=opt //tensorflow/tools/pip_package:build_pip_package ``` , the following failure happens: ``` tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc:76:18: error: 'se' does not name a type using StatusOr = se::port::StatusOr<T>; ^ tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc:139:28: error: 'StatusOr' was not declared in this scope typedef std::function<void(StatusOr<Tensor>)> CommunicationDoneCallback; ... ... ``` This fix addresses the build failures for MPI build. Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/contrib/mpi_collectives')
-rw-r--r--tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc b/tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc
index ed22ee667f..e4b0c2c654 100644
--- a/tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc
+++ b/tensorflow/contrib/mpi_collectives/kernels/mpi_ops.cc
@@ -73,7 +73,7 @@ limitations under the License.
*/
template <class T>
-using StatusOr = se::port::StatusOr<T>;
+using StatusOr = stream_executor::port::StatusOr<T>;
using CPUDevice = Eigen::ThreadPoolDevice;
using GPUDevice = Eigen::GpuDevice;