Skip to content

Commit 925913b

Browse files
narendasannarendasan
committed
some naming stuff
1 parent ebef14e commit 925913b

File tree

27 files changed

+57
-51
lines changed

27 files changed

+57
-51
lines changed

‎.bazelrc‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ build:cxx11_abi --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=1"
3636
build:cxx11_abi --linkopt="-D_GLIBCXX_USE_CXX11_ABI=1"
3737
build:cxx11_abi --define=abi=cxx11_abi
3838

39-
build:jetpack --//toolchains/dep_collection:compute_lib_collection=jetpack
39+
build:jetpack --//toolchains/dep_collection:compute_libs=jetpack
4040

4141
build:ci_testing --define=torchtrt_src=prebuilt --cxxopt="-DDISABLE_TEST_IN_CI" --action_env "NVIDIA_TF32_OVERRIDE=0"
4242
build:use_precompiled_torchtrt --define=torchtrt_src=prebuilt

‎MODULE.bazel‎

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,10 +40,13 @@ new_local_repository(
4040
path = "/usr/local/cuda-12.8",
4141
)
4242

43+
# Server Arm (SBSA) and Jetson Jetpack (L4T) use different versions of CUDA and TensorRT
44+
# These versions can be selected using the flag `--//toolchains/dep_collection:compute_libs="jetpack"`
45+
4346
new_local_repository(
4447
name = "cuda_l4t",
4548
build_file = "@//third_party/cuda:BUILD",
46-
path = "/usr/local/cuda-12.8",
49+
path = "/usr/local/cuda-12.6",
4750
)
4851

4952
new_local_repository(
@@ -75,12 +78,15 @@ http_archive(
7578
)
7679

7780

81+
# It is possible to specify a wheel file to use as the libtorch source by providing the URL below and
82+
# using the build flag `--//toolchains/dep_src:torch="whl"`
83+
7884
http_archive(
7985
name = "torch_whl",
8086
build_file = "@//third_party/libtorch:BUILD",
8187
strip_prefix = "torch",
8288
type = "zip",
83-
urls = ["https://download.pytorch.org/whl/nightly/cu128/torch-2.8.0.dev20250414%2Bcu128-cp39-cp39-manylinux_2_28_aarch64.whl"],
89+
urls = ["https://pypi.jetson-ai-lab.dev/jp6/cu126/+f/52c/2cbdd62b78f32/torch-2.7.0-cp310-cp310-linux_aarch64.whl#sha256=52c2cbdd62b78f32c51fa178212e4721241a2ba9e0c4d7d690dd808bd890d51b"],
8490
)
8591

8692
# Download these tarballs manually from the NVIDIA website

‎core/BUILD‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ config_setting(
1616
"@platforms//cpu:aarch64",
1717
],
1818
flag_values = {
19-
"//toolchains/dep_collection:compute_lib_collection": "default"
19+
"//toolchains/dep_collection:compute_libs": "default"
2020
},
2121
)
2222

@@ -26,7 +26,7 @@ config_setting(
2626
"@platforms//cpu:aarch64",
2727
],
2828
flag_values = {
29-
"//toolchains/dep_collection:compute_lib_collection": "jetpack"
29+
"//toolchains/dep_collection:compute_libs": "jetpack"
3030
},
3131
)
3232

‎core/conversion/BUILD‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ config_setting(
1616
"@platforms//cpu:aarch64",
1717
],
1818
flag_values = {
19-
"//toolchains/dep_collection:compute_lib_collection": "default"
19+
"//toolchains/dep_collection:compute_libs": "default"
2020
},
2121
)
2222

@@ -26,7 +26,7 @@ config_setting(
2626
"@platforms//cpu:aarch64",
2727
],
2828
flag_values = {
29-
"//toolchains/dep_collection:compute_lib_collection": "jetpack"
29+
"//toolchains/dep_collection:compute_libs": "jetpack"
3030
},
3131
)
3232

‎core/conversion/conversionctx/BUILD‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ config_setting(
1616
"@platforms//cpu:aarch64",
1717
],
1818
flag_values = {
19-
"//toolchains/dep_collection:compute_lib_collection": "default"
19+
"//toolchains/dep_collection:compute_libs": "default"
2020
},
2121
)
2222

@@ -26,7 +26,7 @@ config_setting(
2626
"@platforms//cpu:aarch64",
2727
],
2828
flag_values = {
29-
"//toolchains/dep_collection:compute_lib_collection": "jetpack"
29+
"//toolchains/dep_collection:compute_libs": "jetpack"
3030
},
3131
)
3232

‎core/conversion/converters/BUILD‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ config_setting(
1616
"@platforms//cpu:aarch64",
1717
],
1818
flag_values = {
19-
"//toolchains/dep_collection:compute_lib_collection": "default"
19+
"//toolchains/dep_collection:compute_libs": "default"
2020
},
2121
)
2222

@@ -26,7 +26,7 @@ config_setting(
2626
"@platforms//cpu:aarch64",
2727
],
2828
flag_values = {
29-
"//toolchains/dep_collection:compute_lib_collection": "jetpack"
29+
"//toolchains/dep_collection:compute_libs": "jetpack"
3030
},
3131
)
3232

‎core/conversion/evaluators/BUILD‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ config_setting(
1616
"@platforms//cpu:aarch64",
1717
],
1818
flag_values = {
19-
"//toolchains/dep_collection:compute_lib_collection": "default"
19+
"//toolchains/dep_collection:compute_libs": "default"
2020
},
2121
)
2222

@@ -26,7 +26,7 @@ config_setting(
2626
"@platforms//cpu:aarch64",
2727
],
2828
flag_values = {
29-
"//toolchains/dep_collection:compute_lib_collection": "jetpack"
29+
"//toolchains/dep_collection:compute_libs": "jetpack"
3030
},
3131
)
3232

‎core/conversion/tensorcontainer/BUILD‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ config_setting(
1616
"@platforms//cpu:aarch64",
1717
],
1818
flag_values = {
19-
"//toolchains/dep_collection:compute_lib_collection": "default"
19+
"//toolchains/dep_collection:compute_libs": "default"
2020
},
2121
)
2222

@@ -26,7 +26,7 @@ config_setting(
2626
"@platforms//cpu:aarch64",
2727
],
2828
flag_values = {
29-
"//toolchains/dep_collection:compute_lib_collection": "jetpack"
29+
"//toolchains/dep_collection:compute_libs": "jetpack"
3030
},
3131
)
3232

‎core/conversion/var/BUILD‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ config_setting(
1616
"@platforms//cpu:aarch64",
1717
],
1818
flag_values = {
19-
"//toolchains/dep_collection:compute_lib_collection": "default"
19+
"//toolchains/dep_collection:compute_libs": "default"
2020
},
2121
)
2222

@@ -26,7 +26,7 @@ config_setting(
2626
"@platforms//cpu:aarch64",
2727
],
2828
flag_values = {
29-
"//toolchains/dep_collection:compute_lib_collection": "jetpack"
29+
"//toolchains/dep_collection:compute_libs": "jetpack"
3030
},
3131
)
3232

‎core/ir/BUILD‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ config_setting(
1616
"@platforms//cpu:aarch64",
1717
],
1818
flag_values = {
19-
"//toolchains/dep_collection:compute_lib_collection": "default"
19+
"//toolchains/dep_collection:compute_libs": "default"
2020
},
2121
)
2222

@@ -26,7 +26,7 @@ config_setting(
2626
"@platforms//cpu:aarch64",
2727
],
2828
flag_values = {
29-
"//toolchains/dep_collection:compute_lib_collection": "jetpack"
29+
"//toolchains/dep_collection:compute_libs": "jetpack"
3030
},
3131
)
3232

0 commit comments

Comments
 (0)