deepray/WORKSPACE at main · deepray-AI/deepray · 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
124
125
126
127
128
129
130
workspace(name = "deepray")

load("@bazel_tools//tools/build_defs/repo:git.bzl", "git_repository")  # buildifier: disable=load-on-top
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")  # buildifier: disable=load-on-top

http_archive(
    name = "rules_python",
    sha256 = "d71d2c67e0bce986e1c5a7731b4693226867c45bfe0b7c5e0067228a536fc580",
    strip_prefix = "rules_python-0.29.0",
    url = "https://github.com/bazelbuild/rules_python/releases/download/0.29.0/rules_python-0.29.0.tar.gz",
)

load("@rules_python//python:repositories.bzl", "py_repositories", "python_register_toolchains")  # buildifier: disable=load-on-top

py_repositories()

python_register_toolchains(
    name = "python",
    ignore_root_user_error = True,
    python_version = "3.10",
)

load("//third_party/xla:workspace.bzl", xla_repo = "repo")

xla_repo()

# Initialize hermetic Python
load("@xla//third_party/py:python_init_rules.bzl", "python_init_rules")

python_init_rules()

load("@xla//third_party/py:python_init_repositories.bzl", "python_init_repositories")

python_init_repositories(
    default_python_version = "system",
    requirements = {
        "3.10": "//build_deps:requirements_lock_3_10.txt",
        "3.11": "//build_deps:requirements_lock_3_11.txt",
        "3.12": "//build_deps:requirements_lock_3_12.txt",
        "3.13": "//build_deps:requirements_lock_3_13.txt",
    },
)

load("@xla//third_party/py:python_init_toolchains.bzl", "python_init_toolchains")

python_init_toolchains()

load("//third_party/py:python_init_pip.bzl", "python_init_pip")

python_init_pip()

load("@pypi//:requirements.bzl", "install_deps")

install_deps()

load("@xla//:workspace4.bzl", "xla_workspace4")

xla_workspace4()

load("@tsl//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl", "cuda_json_init_repository")

cuda_json_init_repository()

load("@cuda_redist_json//:distributions.bzl", "CUDA_REDISTRIBUTIONS", "CUDNN_REDISTRIBUTIONS")
load("@tsl//third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl", "cuda_redist_init_repositories", "cudnn_redist_init_repository")

cuda_redist_init_repositories(cuda_redistributions = CUDA_REDISTRIBUTIONS)

cudnn_redist_init_repository(cudnn_redistributions = CUDNN_REDISTRIBUTIONS)

load("@tsl//third_party/gpus/cuda/hermetic:cuda_configure.bzl", "cuda_configure")

cuda_configure(name = "local_config_cuda")

load("@tsl//third_party/nccl/hermetic:nccl_redist_init_repository.bzl", "nccl_redist_init_repository")

nccl_redist_init_repository()

load("@tsl//third_party/nccl/hermetic:nccl_configure.bzl", "nccl_configure")

nccl_configure(name = "local_config_nccl")

load("//build_deps/tf_dependency:tf_configure.bzl", "tf_configure")

tf_configure(name = "local_config_tf")

http_archive(
    name = "org_tensorflow",
    patch_args = ["-p1"],
    patches = [
        "//third_party/tf:tf_215.patch",
    ],
    sha256 = "f36416d831f06fe866e149c7cd752da410a11178b01ff5620e9f265511ed57cf",
    strip_prefix = "tensorflow-2.15.1",
    urls = [
        "https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.15.1.tar.gz",
    ],
)

load("@org_tensorflow//tensorflow:workspace3.bzl", "tf_workspace3")

tf_workspace3()

load("@org_tensorflow//tensorflow:workspace2.bzl", "tf_workspace2")

tf_workspace2()

load("@org_tensorflow//tensorflow:workspace1.bzl", "tf_workspace1")

tf_workspace1()

load("@org_tensorflow//tensorflow:workspace0.bzl", "tf_workspace0")

tf_workspace0()

load("@//deepray:workspace3.bzl", "dp_workspace3")

dp_workspace3()

load("@//deepray:workspace2.bzl", "dp_workspace2")

dp_workspace2()

# load("@//deepray:workspace1.bzl", "dp_workspace1")

# dp_workspace1()

load("@//deepray:workspace0.bzl", "dp_workspace0")

dp_workspace0()