Skip to content

Commit 51412c7

Browse files
author
Anurag Dixit
committed
feat: Updating the pre_built to prebuilt
Signed-off-by: Anurag Dixit <[email protected]>
1 parent 5e0880b commit 51412c7

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

docker/dist-test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,5 +5,5 @@ pip3 install timm --trusted-host pypi.org --trusted-host pypi.python.org --trust
55
cd tests/modules && python3 ./hub.py
66
cd ../..
77

8-
bazel test //tests:tests //tests:python_api_tests --compilation_mode=opt --jobs=4 --define=torchtrt_src=pre_built
8+
bazel test //tests:tests //tests:python_api_tests --compilation_mode=opt --jobs=4 --define=torchtrt_src=prebuilt
99

tests/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,10 @@ Currently, the default strategy when we run all the tests (`bazel test //tests`)
2727
In order to **not** build the entire Torch-TensorRT library and only build the test scripts, please use the following command.
2828

2929
```
30-
bazel test //tests --compilation_mode=dbg --test_output=summary --define torchtrt_src=pre_built --jobs 2
30+
bazel test //tests --compilation_mode=dbg --test_output=summary --define torchtrt_src=prebuilt --jobs 2
3131
```
3232

33-
The flag `--define torchtrt_src=pre_built` signals bazel to use pre-compiled library as an external dependency for tests. The pre-compiled library path is defined as a `local_repository` rule in root `WORKSPACE` file (`https://github.com/NVIDIA/Torch-TensorRT/blob/master/WORKSPACE`).
33+
The flag `--define torchtrt_src=prebuilt` signals bazel to use pre-compiled library as an external dependency for tests. The pre-compiled library path is defined as a `local_repository` rule in root `WORKSPACE` file (`https://github.com/NVIDIA/Torch-TensorRT/blob/master/WORKSPACE`).
3434

3535
```
3636
# External dependency for torch_tensorrt if you already have precompiled binaries.

tests/util/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ config_setting(
1010
config_setting(
1111
name = "ci_build_testing",
1212
values = {
13-
"define": "torchtrt_src=pre_built"
13+
"define": "torchtrt_src=prebuilt"
1414
}
1515
)
1616

0 commit comments

Comments
 (0)