@@ -3,33 +3,32 @@ workspace(name = "Torch-TensorRT")
3
3
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
4
4
load("@bazel_tools//tools/build_defs/repo:git.bzl", "git_repository")
5
5
6
- git_repository (
6
+ http_archive (
7
7
name = "rules_python",
8
- remote = "https://github.com/bazelbuild/rules_python.git",
9
- commit = "4fcc24fd8a850bdab2ef2e078b1de337eea751a6",
10
- shallow_since = "1589292086 -0400"
8
+ sha256 = "778197e26c5fbeb07ac2a2c5ae405b30f6cb7ad1f5510ea6fdac03bded96cc6f",
9
+ url = "https://github.com/bazelbuild/rules_python/releases/download/0.2.0/rules_python-0.2.0.tar.gz",
11
10
)
12
11
13
- load("@rules_python//python:repositories.bzl", "py_repositories")
14
- py_repositories()
15
-
16
- load("@rules_python//python:pip.bzl", "pip_repositories", "pip3_import")
17
- pip_repositories()
12
+ load("@rules_python//python:pip.bzl", "pip_install")
18
13
19
14
http_archive(
20
15
name = "rules_pkg",
21
- url = "https://github.com/bazelbuild/rules_pkg/releases/download/0.2.4/rules_pkg-0.2.4.tar.gz",
22
- sha256 = "4ba8f4ab0ff85f2484287ab06c0d871dcb31cc54d439457d28fd4ae14b18450a",
16
+ sha256 = "038f1caa773a7e35b3663865ffb003169c6a71dc995e39bf4815792f385d837d",
17
+ urls = [
18
+ "https://mirror.bazel.build/github.com/bazelbuild/rules_pkg/releases/download/0.4.0/rules_pkg-0.4.0.tar.gz",
19
+ "https://github.com/bazelbuild/rules_pkg/releases/download/0.4.0/rules_pkg-0.4.0.tar.gz",
20
+ ],
23
21
)
24
22
25
23
load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")
24
+
26
25
rules_pkg_dependencies()
27
26
28
27
git_repository(
29
28
name = "googletest",
30
- remote = "https://github.com/google/googletest",
31
29
commit = "703bd9caab50b139428cea1aaff9974ebee5742e",
32
- shallow_since = "1570114335 -0400"
30
+ remote = "https://github.com/google/googletest",
31
+ shallow_since = "1570114335 -0400",
33
32
)
34
33
35
34
# CUDA should be installed on the system locally
@@ -52,17 +51,17 @@ new_local_repository(
52
51
http_archive(
53
52
name = "libtorch",
54
53
build_file = "@//third_party/libtorch:BUILD",
55
- sha256 = "190e963e739d5f7c2dcf94b3994de8fcd335706a4ebb333812ea7d8c841beb06 ",
54
+ sha256 = "8d9e829ce9478db4f35bdb7943308cf02e8a2f58cf9bb10f742462c1d57bf287 ",
56
55
strip_prefix = "libtorch",
57
- urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.10 .0%2Bcu113.zip"],
56
+ urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.11 .0%2Bcu113.zip"],
58
57
)
59
58
60
59
http_archive(
61
60
name = "libtorch_pre_cxx11_abi",
62
61
build_file = "@//third_party/libtorch:BUILD",
63
- sha256 = "0996a6a4ea8bbc1137b4fb0476eeca25b5efd8ed38955218dec1b73929090053 ",
62
+ sha256 = "90159ecce3ff451f3ef3f657493b6c7c96759c3b74bbd70c1695f2ea2f81e1ad ",
64
63
strip_prefix = "libtorch",
65
- urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.10 .0%2Bcu113.zip"],
64
+ urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.11 .0%2Bcu113.zip"],
66
65
)
67
66
68
67
####################################################################################
@@ -84,18 +83,7 @@ new_local_repository(
84
83
#########################################################################
85
84
# Testing Dependencies (optional - comment out on aarch64)
86
85
#########################################################################
87
- pip3_import(
88
- name = "torch_tensorrt_py_deps",
89
- requirements = "//py:requirements.txt"
90
- )
91
-
92
- load("@torch_tensorrt_py_deps//:requirements.bzl", "pip_install")
93
- pip_install()
94
-
95
- pip3_import(
96
- name = "py_test_deps",
97
- requirements = "//tests/py:requirements.txt"
98
- )
99
-
100
- load("@py_test_deps//:requirements.bzl", "pip_install")
101
- pip_install()
86
+ pip_install(
87
+ name = "pylinter_deps",
88
+ requirements = "//tools/linter:requirements.txt",
89
+ )
0 commit comments