-
Notifications
You must be signed in to change notification settings - Fork 53
/
WORKSPACE
45 lines (27 loc) · 1.31 KB
/
WORKSPACE
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
workspace(name = "maga_transformer")
load("//3rdparty/cuda_config:cuda_configure.bzl", "cuda_configure")
load("//3rdparty/gpus:rocm_configure.bzl", "rocm_configure")
load("//3rdparty/py:python_configure.bzl", "python_configure")
cuda_configure(name = "local_config_cuda")
rocm_configure(name = "local_config_rocm")
python_configure(name = "local_config_python")
load("//deps:http.bzl", "http_deps")
http_deps()
load("//deps:git.bzl", "git_deps")
git_deps()
load("@rules_python//python:repositories.bzl", "py_repositories")
py_repositories()
load("//deps:pip.bzl", "pip_deps")
pip_deps()
load("@pip_cpu_torch//:requirements.bzl", pip_cpu_torch_install_deps = "install_deps")
pip_cpu_torch_install_deps()
load("@pip_arm_torch//:requirements.bzl", pip_arm_torch_install_deps = "install_deps")
pip_arm_torch_install_deps()
load("@pip_gpu_torch//:requirements.bzl", pip_gpu_torch_install_deps = "install_deps")
pip_gpu_torch_install_deps()
load("@pip_ppu_torch//:requirements.bzl", pip_ppu_torch_install_deps = "install_deps")
pip_ppu_torch_install_deps()
load("@pip_gpu_cuda12_torch//:requirements.bzl", pip_gpu_cuda12_torch_install_deps = "install_deps")
pip_gpu_cuda12_torch_install_deps()
load("@pip_gpu_rocm_torch//:requirements.bzl", pip_gpu_rocm_torch_install_deps = "install_deps")
pip_gpu_rocm_torch_install_deps()