serving/MODULE.bazel at main · secretflow/serving · GitHub

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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
# Copyright 2025 Ant Group Co., Ltd.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

module(
    name = "sf_serving",
    compatibility_level = 1,
)

bazel_dep(name = "apple_support", version = "1.17.1")
bazel_dep(name = "protobuf", version = "27.3")
bazel_dep(name = "rules_pkg", version = "1.1.0")
bazel_dep(name = "spdlog", version = "1.15.2")
bazel_dep(name = "gflags", version = "2.2.2")
bazel_dep(name = "opentelemetry-proto", version = "1.5.0")
bazel_dep(name = "prometheus-cpp", version = "1.3.0.bcr.2")
bazel_dep(name = "abseil-cpp", version = "20250127.1")
bazel_dep(name = "rapidjson", version = "1.1.0.bcr.20241007")
bazel_dep(name = "pybind11_bazel", version = "2.13.6")
bazel_dep(name = "rules_cuda", version = "0.2.3")
bazel_dep(name = "rules_cc", version = "0.1.1")
bazel_dep(name = "rules_foreign_cc")
bazel_dep(name = "platforms", version = "0.0.11")
bazel_dep(name = "opentelemetry-cpp", version = "1.19.0")

single_version_override(
    module_name = "rules_foreign_cc",
    version = "0.12.0",
)

cuda = use_extension("@rules_cuda//cuda:extensions.bzl", "toolchain")
cuda.local_toolchain(
    name = "local_cuda",
    toolkit_path = "",
)
use_repo(cuda, "local_cuda")

# self-host registry
bazel_dep(name = "yacl", version = "0.4.5b10-nightly-20250110")
bazel_dep(name = "arrow", version = "14.0.2.bcr.1")
bazel_dep(name = "openssl", version = "3.3.2.bcr.1")
bazel_dep(name = "heu", version = "0.6.0.dev20250123")
bazel_dep(name = "kuscia", version = "0.14.0b0")
bazel_dep(name = "dataproxy", version = "0.4.0b0")
bazel_dep(name = "aws_sdk_cpp")
single_version_override(
    module_name = "aws_sdk_cpp",
    patch_strip = 1,
    patches = [
        "//bazel/patches:aws_1.7.336.patch",
    ],
    version = "1.7.336",
)

single_version_override(
    module_name = "grpc",
    version = "1.66.0.bcr.4",
)

bazel_dep(name = "brpc")
single_version_override(
    module_name = "brpc",
    patch_strip = 1,
    patches = [
        # Serving implements the x-b3 header logic using opentelemetry-cpp.
        # Therefore, we use a patch here to revert to the original x-bd header logic of brpc.
        "//bazel/patches:brpc.patch",
    ],
    version = "1.11.0-20241212-282bc90",
)

# python
bazel_dep(name = "rules_python", version = "1.3.0")

DEFAULT_PYTHON_VERSION = "3.10"

SUPPORTED_PYTHON_VERSIONS = [
    "3.9",
    "3.10",
    "3.11",
]

python = use_extension("@rules_python//python/extensions:python.bzl", "python")

[
    python.toolchain(
        configure_coverage_tool = True,
        ignore_root_user_error = True,
        is_default = python_version == DEFAULT_PYTHON_VERSION,
        python_version = python_version,
    )
    for python_version in SUPPORTED_PYTHON_VERSIONS
]

use_repo(python, "python_versions")

pip = use_extension("@rules_python//python/extensions:pip.bzl", "pip")

[
    pip.parse(
        hub_name = "serving_pip",
        python_version = python_version,
        requirements_linux = "//:requirements_lock_{}.txt".format(python_version.replace(".", "_")),
    )
    for python_version in SUPPORTED_PYTHON_VERSIONS
]

use_repo(pip, "serving_pip")
use_repo(pip, "dataproxy_sdk_pip")

# test
bazel_dep(name = "googletest", version = "1.16.0", dev_dependency = True)
bazel_dep(name = "google_benchmark", version = "1.9.2", dev_dependency = True)