From e97c9e91e016efd951dc52e82744f607d948bb2a Mon Sep 17 00:00:00 2001
From: Jianwei Xie <xiejw@google.com>
Date: Wed, 28 Mar 2018 14:36:18 -0700
Subject: Merge changes from github.

PiperOrigin-RevId: 190835392
---
 tensorflow/tensorflow.bzl | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'tensorflow/tensorflow.bzl')

diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl
index 2d3cb415fe..fcc57d506e 100644
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@@ -22,6 +22,7 @@ load(
 load(
     "//third_party/mkl:build_defs.bzl",
     "if_mkl",
+    "if_mkl_lnx_x64"
 )
 
 def register_extension_info(**kwargs):
@@ -202,7 +203,8 @@ def tf_copts(android_optimization_level_override="-O2", is_external=False):
           "-ftemplate-depth=900"])
       + if_cuda(["-DGOOGLE_CUDA=1"])
       + if_tensorrt(["-DGOOGLE_TENSORRT=1"])
-      + if_mkl(["-DINTEL_MKL=1", "-DEIGEN_USE_VML", "-fopenmp",])
+      + if_mkl(["-DINTEL_MKL=1", "-DEIGEN_USE_VML"])
+      + if_mkl_lnx_x64(["-fopenmp"])
       + if_android_arm(["-mfpu=neon"])
       + if_linux_x86_64(["-msse3"])
       + if_ios_x86_64(["-msse4.1"])
-- 
cgit v1.2.3