diff options
Diffstat (limited to 'third_party/mpi/mpi.bzl')
| -rw-r--r-- | third_party/mpi/mpi.bzl | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/third_party/mpi/mpi.bzl b/third_party/mpi/mpi.bzl index 1f7d2f4102..38ce91c4d0 100644 --- a/third_party/mpi/mpi.bzl +++ b/third_party/mpi/mpi.bzl @@ -9,3 +9,9 @@ def mpi_hdr(): else: hdrs = ["mpi.h", "mpio.h", "mpicxx.h"] #When using MVAPICH return hdrs + +def if_mpi(if_true, if_false = []): + return select({ + "//tensorflow:with_mpi_support": if_true, + "//conditions:default": if_false + }) |