diff options
| author |  Tony Wang <tonywy@google.com> | 2018-04-05 17:24:33 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-05 17:26:58 -0700 |
| commit | 165a87a0f938aef649f7e817193716349e42072a (patch) | |
| tree | df7577fc80e9085d4f13996266455f8e11ac710a /tensorflow/compiler/xla/service/cpu/BUILD | |
| parent | cfeadf0986ad60b0ae1eb18b3802539803c63b94 (diff) | |
Automated g4 rollback of changelist 191605505
PiperOrigin-RevId: 191824447
Diffstat (limited to 'tensorflow/compiler/xla/service/cpu/BUILD')
| -rw-r--r-- | tensorflow/compiler/xla/service/cpu/BUILD | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/cpu/BUILD b/tensorflow/compiler/xla/service/cpu/BUILD index 966e2d0fc5..246b802861 100644 --- a/tensorflow/compiler/xla/service/cpu/BUILD +++ b/tensorflow/compiler/xla/service/cpu/BUILD @@ -18,6 +18,10 @@ load(":build_defs.bzl", "runtime_copts") load("//tensorflow:tensorflow.bzl", "tf_cc_test") load("//tensorflow:tensorflow.bzl", "tf_cc_binary") load("//tensorflow/compiler/xla:xla.bzl", "ORC_JIT_MEMORY_MAPPER_TARGETS") +load( + "//third_party/mkl:build_defs.bzl", + "if_mkl", +) # Filegroup used to collect source files for dependency checking. filegroup( @@ -170,6 +174,7 @@ cc_library( ":runtime_fft", ":runtime_fork_join", ":runtime_matmul", + ":runtime_matmul_mkl", ":runtime_single_threaded_conv2d", ":runtime_single_threaded_matmul", "@llvm//:execution_engine", @@ -539,6 +544,22 @@ cc_library( ) cc_library( + name = "runtime_matmul_mkl", + srcs = ["runtime_matmul_mkl.cc"], + hdrs = ["runtime_matmul_mkl.h"], + copts = runtime_copts(), + visibility = ["//visibility:public"], + deps = [ + "//tensorflow/compiler/xla:executable_run_options", + "//tensorflow/core:framework_lite", + "//third_party/eigen3", + ] + if_mkl([ + "//third_party/mkl:intel_binary_blob", + "@mkl_dnn", + ]), +) + +cc_library( name = "runtime_single_threaded_conv2d", srcs = [ "runtime_conv2d_impl.h", @@ -584,10 +605,12 @@ cc_library( tf_cc_test( name = "cpu_runtime_test", srcs = ["cpu_runtime_test.cc"], + shard_count = 10, tags = ["optonly"], deps = [ ":cpu_runtime", ":runtime_matmul", + ":runtime_matmul_mkl", ":runtime_single_threaded_matmul", "//tensorflow/compiler/xla:array2d", "//tensorflow/compiler/xla:types", |