Update tf_configure.bzl to allow centos 7 build

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
yongtang · Feb 15, 2020 · 3ddffb1 · 3ddffb1
1 parent 417febb
commit 3ddffb1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/third_party/toolchains/tf/tf_configure.bzl b/third_party/toolchains/tf/tf_configure.bzl
@@ -156,6 +156,14 @@ def _symlink_genrule_for_dir(
         else:
             dest_files = files.replace(src_dir, "").splitlines()
         src_files = files.splitlines()
+        # Centos may have header files in both lib and lib64:
+        if src_dir.count("/lib64/"):
+            extra_src_dir = src_dir.replace("/lib64/", "/lib/")
+            extra_files = "\n".join(sorted(_read_dir(repository_ctx, extra_src_dir).splitlines()))
+            extra_dest_files = extra_files.replace(extra_src_dir, "").splitlines()
+            extra_src_files = extra_files.splitlines()
+            src_files = src_files + extra_src_files
+            dest_files = dest_files + extra_dest_files
     command = []
     outs = []