1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
diff --git a/third_party/mkl/build_defs.bzl b/third_party/mkl/build_defs.bzl
index 4b8fb83eb0..f4e1adfb22 100644
--- a/third_party/mkl/build_defs.bzl
+++ b/third_party/mkl/build_defs.bzl
@@ -124,7 +124,7 @@ def _mkl_autoconf_impl(repository_ctx):
if _enable_local_mkl(repository_ctx):
# Symlink lib and include local folders.
mkl_root = repository_ctx.os.environ[_TF_MKL_ROOT]
- mkl_lib_path = "%s/lib" % mkl_root
+ mkl_lib_path = "%s/lib/intel64" % mkl_root
repository_ctx.symlink(mkl_lib_path, "lib")
mkl_include_path = "%s/include" % mkl_root
repository_ctx.symlink(mkl_include_path, "include")
diff --git a/third_party/mkl/mkl.BUILD b/third_party/mkl/mkl.BUILD
index 72370182c4..4972bb005e 100644
--- a/third_party/mkl/mkl.BUILD
+++ b/third_party/mkl/mkl.BUILD
@@ -5,7 +5,6 @@ exports_files(["license.txt"])
filegroup(
name = "LICENSE",
srcs = [
- "license.txt",
],
visibility = ["//visibility:public"],
)
@@ -21,7 +20,10 @@ cc_library(
name = "mkl_libs_linux",
srcs = [
"lib/libiomp5.so",
- "lib/libmklml_intel.so",
+ "lib/libmkl_core.so",
+ "lib/libmkl_rt.so",
+ "lib/libmkl_intel_thread.so",
+ "lib/libmkl_intel_lp64.so",
],
visibility = ["//visibility:public"],
)
|