diff options
author | Yong Tang <yong.tang.github@outlook.com> | 2018-04-26 20:21:18 +0000 |
---|---|---|
committer | Yong Tang <yong.tang.github@outlook.com> | 2018-04-26 20:21:18 +0000 |
commit | 26b2814096bf6d9dd0af91a37a0706e17450b9ec (patch) | |
tree | 6a512bde923df559c1e54f34053315530f463ef8 /tensorflow/contrib/mpi | |
parent | 291d85be428b63c72fadf432ba7af9a0d635e52f (diff) |
Fix build error with MPI support
This fix tries to fix the issue raised in 18363 where
the bazel build with MPI support fails as a header is missing
in the include.
This fix fixes the issue. The fix is verified locally
with MPI+CUDA on Ubuntu 16.04.
This fix fixes 18363.
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/contrib/mpi')
-rw-r--r-- | tensorflow/contrib/mpi/mpi_utils.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/mpi/mpi_utils.h b/tensorflow/contrib/mpi/mpi_utils.h index df055ff567..4091925fc0 100644 --- a/tensorflow/contrib/mpi/mpi_utils.h +++ b/tensorflow/contrib/mpi/mpi_utils.h @@ -22,6 +22,7 @@ limitations under the License. #include <string> #include <vector> +#include "tensorflow/core/platform/logging.h" #include "tensorflow/core/lib/strings/str_util.h" // Skip MPI C++ bindings support, this matches the usage in other places |