add bazel tools to generate tao build version info

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")

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
+    ],
+)

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}"