72 lines
3.4 KiB
Python
72 lines
3.4 KiB
Python
load("//third_party/gpus:cuda_configure.bzl", "cuda_configure")
|
|
load("//third_party/gpus:rocm_configure.bzl", "rocm_configure")
|
|
load("//third_party/llvm:workspace.bzl", llvm = "repo")
|
|
load("//third_party/py:python_repo.bzl", "python_repository")
|
|
load("//third_party/pybind11_bazel:workspace.bzl", pybind11_bazel = "repo")
|
|
load("//third_party/stablehlo:workspace.bzl", stablehlo = "repo")
|
|
load("//third_party/tensorrt:tensorrt_configure.bzl", "tensorrt_configure")
|
|
load("//third_party/triton:workspace.bzl", triton = "repo")
|
|
load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
|
|
load("//third_party:repo.bzl", "tf_vendored")
|
|
load("//tools/toolchains/remote:configure.bzl", "remote_execution_configure")
|
|
|
|
def _workspace_private_impl(mctx):
|
|
cuda_configure(name = "local_config_cuda")
|
|
remote_execution_configure(name = "local_config_remote_execution")
|
|
rocm_configure(name = "local_config_rocm")
|
|
tensorrt_configure(name = "local_config_tensorrt")
|
|
tf_vendored(name = "tsl", relpath = "third_party/tsl")
|
|
pybind11_bazel()
|
|
tf_http_archive(
|
|
name = "com_github_grpc_grpc",
|
|
sha256 = "b956598d8cbe168b5ee717b5dafa56563eb5201a947856a6688bbeac9cac4e1f",
|
|
strip_prefix = "grpc-b54a5b338637f92bfcf4b0bc05e0f57a5fd8fadd",
|
|
system_build_file = "//third_party/systemlibs:grpc.BUILD",
|
|
patch_file = [
|
|
"//third_party/grpc:generate_cc_env_fix.patch",
|
|
"//third_party/grpc:register_go_toolchain.patch",
|
|
],
|
|
system_link_files = {
|
|
"//third_party/systemlibs:BUILD.bazel": "bazel/BUILD.bazel",
|
|
"//third_party/systemlibs:grpc.BUILD": "src/compiler/BUILD",
|
|
"//third_party/systemlibs:grpc.bazel.grpc_deps.bzl": "bazel/grpc_deps.bzl",
|
|
"//third_party/systemlibs:grpc.bazel.grpc_extra_deps.bzl": "bazel/grpc_extra_deps.bzl",
|
|
"//third_party/systemlibs:grpc.bazel.cc_grpc_library.bzl": "bazel/cc_grpc_library.bzl",
|
|
"//third_party/systemlibs:grpc.bazel.generate_cc.bzl": "bazel/generate_cc.bzl",
|
|
"//third_party/systemlibs:grpc.bazel.protobuf.bzl": "bazel/protobuf.bzl",
|
|
},
|
|
urls = tf_mirror_urls("https://github.com/grpc/grpc/archive/b54a5b338637f92bfcf4b0bc05e0f57a5fd8fadd.tar.gz"),
|
|
)
|
|
tf_http_archive(
|
|
name = "com_google_protobuf",
|
|
patch_file = ["//third_party/protobuf:protobuf.patch"],
|
|
sha256 = "f66073dee0bc159157b0bd7f502d7d1ee0bc76b3c1eac9836927511bdc4b3fc1",
|
|
strip_prefix = "protobuf-3.21.9",
|
|
system_build_file = "//third_party/systemlibs:protobuf.BUILD",
|
|
system_link_files = {
|
|
"//third_party/systemlibs:protobuf.bzl": "protobuf.bzl",
|
|
"//third_party/systemlibs:protobuf_deps.bzl": "protobuf_deps.bzl",
|
|
},
|
|
urls = tf_mirror_urls("https://github.com/protocolbuffers/protobuf/archive/v3.21.9.zip"),
|
|
)
|
|
python_repository(
|
|
name = "python_version_repo",
|
|
requirements_versions = ["3.11"],
|
|
requirements_locks = ["//:requirements_lock_3_11.txt"],
|
|
local_wheel_workspaces = [],
|
|
local_wheel_dist_folder = None,
|
|
default_python_version = None,
|
|
local_wheel_inclusion_list = ["*"],
|
|
local_wheel_exclusion_list = [],
|
|
)
|
|
|
|
return mctx.extension_metadata(
|
|
reproducible = True,
|
|
root_module_direct_deps = "all",
|
|
root_module_direct_dev_deps = [],
|
|
)
|
|
|
|
workspace_private = module_extension(
|
|
implementation = _workspace_private_impl,
|
|
)
|