aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tensorflow.bzl
diff options
context:
space:
mode:
authorGravatar Avijit <Avijit.Chakraborty@intel.com>2018-07-25 01:08:01 -0700
committerGravatar Avijit <Avijit.Chakraborty@intel.com>2018-07-25 01:08:01 -0700
commit1cdacb8b10d0b4687387be5fd8be978d68602a1d (patch)
treea2bf88798854a426f073325eb85d85b3ab914418 /tensorflow/tensorflow.bzl
parentf88a6f93bee89c610fa8b399d037c7a33c1a0a3e (diff)
parent3f454e4060d855f43eebe0cdc27d8c24f906d430 (diff)
Merge remote-tracking branch 'upstream/master'
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r--tensorflow/tensorflow.bzl26
1 files changed, 22 insertions, 4 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl
index 5884870daa..6f1a576c6c 100644
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@@ -25,6 +25,10 @@ load(
"if_mkl_lnx_x64"
)
load(
+ "//third_party/mkl_dnn:build_defs.bzl",
+ "if_mkl_open_source_only",
+)
+load(
"//third_party/ngraph:build_defs.bzl",
"if_ngraph",
)
@@ -138,6 +142,14 @@ def if_not_mobile(a):
"//conditions:default": a,
})
+# Config setting selector used when building for products
+# which requires restricted licenses to be avoided.
+def if_not_lgpl_restricted(a):
+ _ = (a,)
+ return select({
+ "//conditions:default": [],
+ })
+
def if_not_windows(a):
return select({
clean_dep("//tensorflow:windows"): [],
@@ -184,9 +196,13 @@ def get_win_copts(is_external=False):
"/DEIGEN_AVOID_STL_ARRAY",
"/Iexternal/gemmlowp",
"/wd4018", # -Wno-sign-compare
- "/U_HAS_EXCEPTIONS",
- "/D_HAS_EXCEPTIONS=1",
- "/EHsc", # -fno-exceptions
+ # Bazel's CROSSTOOL currently pass /EHsc to enable exception by
+ # default. We can't pass /EHs-c- to disable exception, otherwise
+ # we will get a waterfall of flag conflict warnings. Wait for
+ # Bazel to fix this.
+ # "/D_HAS_EXCEPTIONS=0",
+ # "/EHs-c-",
+ "/wd4577",
"/DNOGDI",
]
if is_external:
@@ -218,6 +234,7 @@ def tf_copts(android_optimization_level_override="-O2", is_external=False):
+ if_cuda(["-DGOOGLE_CUDA=1"])
+ if_tensorrt(["-DGOOGLE_TENSORRT=1"])
+ if_mkl(["-DINTEL_MKL=1", "-DEIGEN_USE_VML"])
+ + if_mkl_open_source_only(["-DDO_NOT_USE_ML"])
+ if_ngraph(["-DINTEL_NGRAPH=1"])
+ if_mkl_lnx_x64(["-fopenmp"])
+ if_android_arm(["-mfpu=neon"])
@@ -233,6 +250,7 @@ def tf_copts(android_optimization_level_override="-O2", is_external=False):
clean_dep("//tensorflow:windows"): get_win_copts(is_external),
clean_dep("//tensorflow:windows_msvc"): get_win_copts(is_external),
clean_dep("//tensorflow:ios"): ["-std=c++11"],
+ clean_dep("//tensorflow:no_lgpl_deps"): ["-D__TENSORFLOW_NO_LGPL_DEPS__", "-pthread"],
"//conditions:default": ["-pthread"]
}))
@@ -1348,7 +1366,7 @@ def tf_custom_op_library(name, srcs=[], gpu_srcs=[], deps=[], linkopts=[]):
name=name,
srcs=srcs,
deps=deps + if_cuda(cuda_deps),
- data=[name + "_check_deps"],
+ data=if_static([name + "_check_deps"]),
copts=tf_copts(is_external=True),
features = ["windows_export_all_symbols"],
linkopts=linkopts + select({