diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2018-07-10 11:59:28 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-10 12:02:59 -0700 |
| commit | 5278b8509e2cd1b2847315db46fc0f958824cfce (patch) | |
| tree | 17194ed977d50f3839733c30c848bcea7389c267 /tensorflow/tensorflow.bzl | |
| parent | 328db40921708a16d8ddc7c54c6f7929955fc62b (diff) | |
Fix compilation of mkl kernels.
PiperOrigin-RevId: 203987173
Diffstat (limited to 'tensorflow/tensorflow.bzl')
| -rw-r--r-- | tensorflow/tensorflow.bzl | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 23fea587b4..e4241667ad 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -825,6 +825,9 @@ def tf_cc_test_mkl(srcs, tags=[], size="medium", args=None): + # -fno-exceptions in nocopts breaks compilation if header modules are enabled. + disable_header_modules = ["-use_header_modules"] + for src in srcs: native.cc_test( name=src_to_test_name(src), @@ -850,6 +853,7 @@ def tf_cc_test_mkl(srcs, tags=tags, size=size, args=args, + features=disable_header_modules, nocopts="-fno-exceptions") @@ -1089,6 +1093,9 @@ def tf_mkl_kernel_library(name, hdrs = hdrs + native.glob( [prefix + "*.h"]) + # -fno-exceptions in nocopts breaks compilation if header modules are enabled. + disable_header_modules = ["-use_header_modules"] + native.cc_library( name=name, srcs=if_mkl(srcs), @@ -1096,7 +1103,8 @@ def tf_mkl_kernel_library(name, deps=deps, alwayslink=alwayslink, copts=copts, - nocopts=nocopts + nocopts=nocopts, + features = disable_header_modules ) register_extension_info( |