diff options
author | 2018-06-20 02:12:08 +0000 | |
---|---|---|
committer | 2018-06-28 15:47:05 +0000 | |
commit | 1b79572c2d47c1b7d82c670c27b9641335b7c025 (patch) | |
tree | 16e8056c8bc383275c05d6a3e1b9f0209fb1f77a /tensorflow/contrib/mpi_collectives | |
parent | 51ddb66cdf1444998827e03c4b5f592841ee6255 (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.cc | 2 |
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; |