From 6896a74984efb4b1b77fc36ea274703536ba649d Mon Sep 17 00:00:00 2001
From: Jon Triebenbach <jlt@us.ibm.com>
Date: Wed, 27 Jun 2018 13:29:53 -0500
Subject: Build OpenBLAS 0.3.0 on ppc64le for TF tests

---
 configure.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'configure.py')

diff --git a/configure.py b/configure.py
index ad585fa52e..04ad1c9441 100644
--- a/configure.py
+++ b/configure.py
@@ -1465,6 +1465,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',
-- 
cgit v1.2.3