-
Notifications
You must be signed in to change notification settings - Fork 0
/
WORKSPACE
59 lines (49 loc) · 1.95 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
56
57
58
59
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
# To update TensorFlow to a new revision,
# a) update URL and strip_prefix to the new git commit hash
# b) get the sha256 hash of the commit by running:
# curl -L https://github.com/tensorflow/tensorflow/archive/<git hash>.tar.gz | sha256sum
# and update the sha256 with the result.
#http_archive(
# name = "org_tensorflow",
# patch_args = [
# "-l",
# "-p1",
# ],
# patch_tool = "patch",
# patches = [
# "//tf_patches:cache_urls.diff",
# "//tf_patches:cudnn_int8x32.diff",
# "//tf_patches:f16_abi_clang.diff",
# "//tf_patches:gpu_race_condition.diff",
# "//tf_patches:grpc_version.diff",
# "//tf_patches:stream_executor.diff",
# "//tf_patches:thread_local_random.diff",
# "//tf_patches:xplane.diff",
# ],
# strip_prefix = "tensorflow-f7759359f8420d3ca7b9fd19493f2a01bd47b4ef",
# urls = [
# "https://github.com/tensorflow/tensorflow/archive/f7759359f8420d3ca7b9fd19493f2a01bd47b4ef.tar.gz",
# ],
#)
# For development, one often wants to make changes to the TF repository as well
# as the PyTorch/XLA repository. You can override the pinned repository above with a
# local checkout by either:
# a) overriding the TF repository on the build.py command line by passing a flag
# like:
# bazel --override_repository=org_tensorflow=/path/to/tensorflow
# or
# b) by commenting out the http_archive above and uncommenting the following:
local_repository(
name = "org_tensorflow",
path = "third_party/tensorflow",
)
# Initialize TensorFlow's external dependencies.
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()