Skip to content

Commit 7ca3ee6

Browse files
committed
adding torchTRT testing dependancy and modifying to py3.12
1 parent efe66fb commit 7ca3ee6

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

docker/MODULE.bazel.ngc

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,12 @@ git_override(
2424

2525
local_repository = use_repo_rule("@bazel_tools//tools/build_defs/repo:local.bzl", "local_repository")
2626

27+
# External dependency for torch_tensorrt if you already have precompiled binaries.
28+
local_repository(
29+
name = "torch_tensorrt",
30+
path = "/usr/local/lib/python3.12/dist-packages/torch_tensorrt/",
31+
)
32+
2733
new_local_repository = use_repo_rule("@bazel_tools//tools/build_defs/repo:local.bzl", "new_local_repository")
2834

2935

@@ -50,13 +56,13 @@ http_archive = use_repo_rule("@bazel_tools//tools/build_defs/repo:http.bzl", "ht
5056

5157
new_local_repository(
5258
name = "libtorch",
53-
path = "/usr/local/lib/python3.10/dist-packages/torch/",
59+
path = "/usr/local/lib/python3.12/dist-packages/torch/",
5460
build_file = "third_party/libtorch/BUILD"
5561
)
5662

5763
new_local_repository(
5864
name = "libtorch_pre_cxx11_abi",
59-
path = "/usr/local/lib/python3.10/dist-packages/torch/",
65+
path = "/usr/local/lib/python3.12/dist-packages/torch/",
6066
build_file = "third_party/libtorch/BUILD"
6167
)
6268

0 commit comments

Comments
 (0)