-
Notifications
You must be signed in to change notification settings - Fork 8
/
WORKSPACE
55 lines (36 loc) · 1.42 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
46
47
48
49
50
51
52
53
54
55
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
load("//build_deps/tf_dependency:tf_configure.bzl", "tf_configure")
load("//build_deps/toolchains/gpu:cuda_configure.bzl", "cuda_configure")
tf_configure(
name = "local_config_tf",
)
http_archive(
name = "org_tensorflow",
strip_prefix = "tensorflow-2.9.1",
urls = [
"https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.9.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()
# Initialize the TensorFlow repository and all dependencies.
#
# The cascade of load() statements and tf_workspace?() calls works around the
# restriction that load() statements need to be at the top of .bzl files.
# E.g. we can not retrieve a new repository with http_archive and then load()
# a macro from that repository in the same file.
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()
cuda_configure(name = "local_config_cuda")