-
Notifications
You must be signed in to change notification settings - Fork 49
/
WORKSPACE_WITH_TF
67 lines (59 loc) · 2.6 KB
/
WORKSPACE_WITH_TF
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
# Workspace configuration for Yggdrasil Decision Forest with dependency to
# TensorFlow (filesystem IO and dataset format). To compile YDF without any
# TensorFlow dependency, use the "WORKSPACE_NO_TF" workspace. See
# tools/test_bazel_no_tf.sh for more details.
workspace(name = "yggdrasil_decision_forests")
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
# To reduce the risk of version mismatch between TensorFlow and
# the other dependencies used by YDF, we don't reimport dependencies
# also imported by TensorFlow.
# Load third party libraries
load("//third_party/gtest:workspace.bzl", gtest = "deps")
# load("//third_party/absl:workspace.bzl", absl = "deps")
# load("//third_party/protobuf:workspace.bzl", protobuf = "deps")
load("//third_party/zlib:workspace.bzl", zlib = "deps")
load("//third_party/tensorflow:workspace.bzl", tensorflow = "deps")
#load("//third_party/farmhash:workspace.bzl", farmhash = "deps")
load("//third_party/boost_math:workspace.bzl", boost_math = "deps")
# load("//third_party/grpc:workspace.bzl", grpc = "deps")
load("//third_party/rapidjson:workspace.bzl", rapidjson = "deps")
load("//third_party/eigen3:workspace.bzl", eigen = "deps")
gtest()
# absl() # We use the abseil linked in tensorflow to avoid package clashes
# protobuf() # We use the protobuf linked in tensorflow.
zlib()
tensorflow()
#farmhash()
boost_math()
# grpc() # We use the protobuf linked in tensorflow.
rapidjson()
# eigen()
# The initialization of YDF dependencies is commented. TensorFlow
# is in charge of initializing them.
# load("@com_google_protobuf//:protobuf_deps.bzl", "protobuf_deps")
# protobuf_deps()
# load("@com_github_grpc_grpc//bazel:grpc_deps.bzl", "grpc_deps")
# grpc_deps()
# load("@com_github_grpc_grpc//bazel:grpc_extra_deps.bzl", "grpc_extra_deps")
# grpc_extra_deps()
# TensorFlow is used to read and write TFRecord and IO if
# use_tensorflow_io=1. Only a small fraction of TF will be compiled.
load("@org_tensorflow//tensorflow:workspace3.bzl", tf1="workspace")
tf1()
load("@org_tensorflow//tensorflow:workspace2.bzl", tf2="workspace")
tf2()
load("@org_tensorflow//tensorflow:workspace1.bzl", tf3="workspace")
tf3()
load("@org_tensorflow//tensorflow:workspace0.bzl", tf4="workspace")
tf4()
# Emscripten
http_archive(
name = "emsdk",
strip_prefix = "emsdk-3.1.15/bazel",
url = "https://github.com/emscripten-core/emsdk/archive/refs/tags/3.1.15.zip",
sha256 = "bc06feb66e376f0184929e603d7c02dcd07646ed6f89281bf1478bb8947fbb0f",
)
load("@emsdk//:deps.bzl", emsdk_deps = "deps")
emsdk_deps()
load("@emsdk//:emscripten_deps.bzl", emsdk_emscripten_deps = "emscripten_deps")
emsdk_emscripten_deps()