diff options
author | 2018-03-08 00:23:00 -0800 | |
---|---|---|
committer | 2018-03-08 00:23:00 -0800 | |
commit | f73d7c90ed05bcf9f36f6a3be0c29efa5fef0f6e (patch) | |
tree | 7ff4938ca96a6e55198ed9627d7c8260202060bd /tensorflow/contrib/mpi | |
parent | f8363dc424f78ec06c9fe2faee7623624aa0392e (diff) |
Add missing `#define OMPI_SKIP_MPICXX` for consistency (#17414)
This fix adds the missing `#define OMPI_SKIP_MPICXX` in
`tensorflow/contrib/mpi/mpi_utils.h` so that it is consistent
with other usages of `mpi.h` includes. `OMPI_SKIP_MPICXX` skip
the MPI C++ bindings support.
This fix fixes 17388.
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 | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/mpi/mpi_utils.h b/tensorflow/contrib/mpi/mpi_utils.h index fa297c28cb..df055ff567 100644 --- a/tensorflow/contrib/mpi/mpi_utils.h +++ b/tensorflow/contrib/mpi/mpi_utils.h @@ -24,6 +24,8 @@ limitations under the License. #include "tensorflow/core/lib/strings/str_util.h" +// Skip MPI C++ bindings support, this matches the usage in other places +#define OMPI_SKIP_MPICXX #include "third_party/mpi/mpi.h" #define MPI_CHECK(cmd) \ do { \ |