diff options
| author |  Jeroen Bédorf <jeroen@minds.ai> | 2017-05-24 01:40:23 -0700 |
|---|---|---|
| committer | Jeroen Bédorf <jeroen@minds.ai> | 2017-05-24 01:40:23 -0700 |
| commit | 047546d6fee2549c5963a338cdcc2ac801097d76 (patch) | |
| tree | c30059df2be9e1fa92e61b72207461fe147acfba /third_party/mpi/mpi.bzl | |
| parent | 8ef441b150c6e309cb73f7b68cf91031a3d3e28b (diff) | |
Fix 'bazel test' errors
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 + }) |