forked from liuliu/ccv
-
Notifications
You must be signed in to change notification settings - Fork 0
/
WORKSPACE
49 lines (41 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
46
47
48
49
workspace(name = "ccv")
load("@bazel_tools//tools/build_defs/repo:git.bzl", "git_repository")
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
git_repository(
name = "bazel_skylib",
remote = "https://github.com/bazelbuild/bazel-skylib.git",
commit = "528e4241345536c487cca8b11db138104bb3bd68",
shallow_since = "1601067301 +0200"
)
git_repository(
name = "build_bazel_rules_cuda",
remote = "https://github.com/liuliu/rules_cuda.git",
commit = "d27ed9b54c3fa1639c20a72550258594f449cbff",
shallow_since = "1618438243 -0400"
)
http_archive(
name = "sqlite3",
sha256 = "b34f4c0c0eefad9a7e515c030c18702e477f4ef7d8ade6142bdab8011b487ac6",
urls = ["https://www.sqlite.org/2020/sqlite-amalgamation-3330000.zip"],
build_file = "sqlite3.BUILD"
)
load("@build_bazel_rules_cuda//gpus:cuda_configure.bzl", "cuda_configure")
load("@build_bazel_rules_cuda//nccl:nccl_configure.bzl", "nccl_configure")
cuda_configure(name = "local_config_cuda")
nccl_configure(name = "local_config_nccl")
load("//config:ccv.bzl", "ccv_setting")
ccv_setting(
name = "local_config_ccv",
have_cblas = True,
have_libpng = True,
have_libjpeg = True,
have_fftw3 = True,
have_pthread = True,
have_liblinear = True,
have_tesseract = True,
have_gsl = True,
have_cudnn = True,
have_nccl = True,
use_openmp = True,
use_dispatch = True
)