aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/mpi
diff options
context:
space:
mode:
authorGravatar Yong Tang <yong.tang.github@outlook.com>2018-03-08 00:23:00 -0800
committerGravatar Gunhan Gulsoy <gunan@google.com>2018-03-08 00:23:00 -0800
commitf73d7c90ed05bcf9f36f6a3be0c29efa5fef0f6e (patch)
tree7ff4938ca96a6e55198ed9627d7c8260202060bd /tensorflow/contrib/mpi
parentf8363dc424f78ec06c9fe2faee7623624aa0392e (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.h2
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 { \