diff options
Diffstat (limited to 'configure.py')
| -rw-r--r-- | configure.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/configure.py b/configure.py index 10387493b0..eaff83d2cc 100644 --- a/configure.py +++ b/configure.py @@ -1459,6 +1459,13 @@ def main(): environ_cp['TF_NEED_JEMALLOC'] = '0' environ_cp['TF_NEED_TENSORRT'] = '0' + # The numpy package on ppc64le uses OpenBLAS which has multi-threading + # issues that lead to incorrect answers. Set OMP_NUM_THREADS=1 at + # runtime to allow the Tensorflow testcases which compare numpy + # results to Tensorflow results to succeed. + if is_ppc64le(): + write_action_env_to_bazelrc("OMP_NUM_THREADS", 1) + set_build_var(environ_cp, 'TF_NEED_JEMALLOC', 'jemalloc as malloc', 'with_jemalloc', True) set_build_var(environ_cp, 'TF_NEED_GCP', 'Google Cloud Platform', |