From 723f2ef256a3fd03dbcc975d38928b07dbd73dac Mon Sep 17 00:00:00 2001
From: "lanbo.llb" <lanbo.llb@alibaba-inc.com>
Date: Tue, 22 Mar 2022 16:52:02 +0800
Subject: [PATCH] add bazel tools to generate tao build version info

---
 tensorflow/workspace2.bzl                     |  2 ++
 third_party/blade_helper/BUILD                |  0
 third_party/blade_helper/BUILD.tpl            |  0
 .../blade_helper/blade_helper_configure.bzl   | 34 +++++++++++++++++++
 third_party/blade_helper/build_defs.bzl.tpl   | 17 ++++++++++
 5 files changed, 53 insertions(+)
 create mode 100644 third_party/blade_helper/BUILD
 create mode 100644 third_party/blade_helper/BUILD.tpl
 create mode 100644 third_party/blade_helper/blade_helper_configure.bzl
 create mode 100644 third_party/blade_helper/build_defs.bzl.tpl

diff --git a/tensorflow/workspace2.bzl b/tensorflow/workspace2.bzl
index 7282f0248f1..845bf08db63 100644
--- a/tensorflow/workspace2.bzl
+++ b/tensorflow/workspace2.bzl
@@ -3,6 +3,7 @@
 # Import third party config rules.
 load("//tensorflow:version_check.bzl", "check_bazel_version_at_least")
 load("//third_party/gpus:cuda_configure.bzl", "cuda_configure")
+load("//third_party/blade_helper:blade_helper_configure.bzl", "blade_helper_configure")
 load("//third_party/gpus:rocm_configure.bzl", "rocm_configure")
 load("//third_party/tensorrt:tensorrt_configure.bzl", "tensorrt_configure")
 load("//third_party/nccl:nccl_configure.bzl", "nccl_configure")
@@ -94,6 +95,7 @@ def _tf_toolchains():
     clang6_configure(name = "local_config_clang6")
     cc_download_clang_toolchain(name = "local_config_download_clang")
     cuda_configure(name = "local_config_cuda")
+    blade_helper_configure(name = "local_config_blade_helper")
     tensorrt_configure(name = "local_config_tensorrt")
     nccl_configure(name = "local_config_nccl")
     git_configure(name = "local_config_git")
diff --git a/third_party/blade_helper/BUILD b/third_party/blade_helper/BUILD
new file mode 100644
index 00000000000..e69de29bb2d
diff --git a/third_party/blade_helper/BUILD.tpl b/third_party/blade_helper/BUILD.tpl
new file mode 100644
index 00000000000..e69de29bb2d
diff --git a/third_party/blade_helper/blade_helper_configure.bzl b/third_party/blade_helper/blade_helper_configure.bzl
new file mode 100644
index 00000000000..dbb263d89ef
--- /dev/null
+++ b/third_party/blade_helper/blade_helper_configure.bzl
@@ -0,0 +1,34 @@
+load("//third_party/remote_config:common.bzl", "get_host_environ")
+
+_TAO_BUILD_VERSION = "TAO_BUILD_VERSION"
+_TAO_BUILD_GIT_BRANCH = "TAO_BUILD_GIT_BRANCH"
+_TAO_BUILD_GIT_HEAD = "TAO_BUILD_GIT_HEAD"
+_TAO_BUILD_HOST = "TAO_BUILD_HOST"
+_TAO_BUILD_IP = "TAO_BUILD_IP"
+_TAO_BUILD_TIME = "TAO_BUILD_TIME"
+
+
+def _blade_helper_impl(repository_ctx):
+    repository_ctx.template("build_defs.bzl", Label("//bazel/blade_helper:build_defs.bzl.tpl"), {
+        "%{TAO_BUILD_VERSION}": get_host_environ(repository_ctx, _TAO_BUILD_VERSION),
+        "%{TAO_BUILD_GIT_BRANCH}": get_host_environ(repository_ctx, _TAO_BUILD_GIT_BRANCH),
+        "%{TAO_BUILD_GIT_HEAD}": get_host_environ(repository_ctx, _TAO_BUILD_GIT_HEAD),
+        "%{TAO_BUILD_HOST}": get_host_environ(repository_ctx, _TAO_BUILD_HOST),
+        "%{TAO_BUILD_IP}": get_host_environ(repository_ctx, _TAO_BUILD_IP),
+        "%{TAO_BUILD_TIME}": get_host_environ(repository_ctx, _TAO_BUILD_TIME),
+    })
+
+    repository_ctx.template("BUILD", Label("//bazel/blade_helper:BUILD.tpl"), {
+    })
+
+blade_helper_configure = repository_rule(
+    implementation = _blade_helper_impl,
+    environ = [
+        _TAO_BUILD_VERSION,
+        _TAO_BUILD_GIT_BRANCH,
+        _TAO_BUILD_GIT_HEAD,
+        _TAO_BUILD_HOST,
+        _TAO_BUILD_IP,
+        _TAO_BUILD_TIME
+    ],
+)
diff --git a/third_party/blade_helper/build_defs.bzl.tpl b/third_party/blade_helper/build_defs.bzl.tpl
new file mode 100644
index 00000000000..6847a22c9de
--- /dev/null
+++ b/third_party/blade_helper/build_defs.bzl.tpl
@@ -0,0 +1,17 @@
+def tao_build_version():
+    return "%{TAO_BUILD_VERSION}"
+
+def tao_build_git_branch():
+    return "%{TAO_BUILD_GIT_BRANCH}"
+
+def tao_build_git_head():
+    return "%{TAO_BUILD_GIT_HEAD}"
+
+def tao_build_host():
+    return "%{TAO_BUILD_HOST}"
+
+def tao_build_ip():
+    return "%{TAO_BUILD_IP}"
+
+def tao_build_time():
+    return "%{TAO_BUILD_TIME}"