diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-16 22:23:03 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-16 22:23:31 -0700 |
commit | a013474eae601ae27c406e6c8da34ef762e8f762 (patch) | |
tree | d129cf87b48b09fbf6dc48b23e6bd648f4d6f173 /configure.py | |
parent | 1a2af489b1087eb22ec76863867e4e397e453e34 (diff) | |
parent | 54addf8f9215bd1142d9fa073d6c1c029a341ef6 (diff) |
Merge pull request #20357 from JonTriebenbach:openblas-for-ppc64le
PiperOrigin-RevId: 204859562
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', |