Skip to content

Commit

Permalink
Merge branch 'anuragd/change_pre_built' into 'release/1.0'
Browse files Browse the repository at this point in the history
feat: Updating the pre_built to prebuilt

See merge request adlsa/TRTorch!14
  • Loading branch information
narendasan committed Nov 2, 2021
2 parents 5e0880b + 51412c7 commit 540e135
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion docker/dist-test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -5,5 +5,5 @@ pip3 install timm --trusted-host pypi.org --trusted-host pypi.python.org --trust
cd tests/modules && python3 ./hub.py
cd ../..

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

4 changes: 2 additions & 2 deletions tests/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -27,10 +27,10 @@ Currently, the default strategy when we run all the tests (`bazel test //tests`)
In order to **not** build the entire Torch-TensorRT library and only build the test scripts, please use the following command.

```
bazel test //tests --compilation_mode=dbg --test_output=summary --define torchtrt_src=pre_built --jobs 2
bazel test //tests --compilation_mode=dbg --test_output=summary --define torchtrt_src=prebuilt --jobs 2
```

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`).
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`).

```
# External dependency for torch_tensorrt if you already have precompiled binaries.
Expand Down
2 changes: 1 addition & 1 deletion tests/util/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ config_setting(
config_setting(
name = "ci_build_testing",
values = {
"define": "torchtrt_src=pre_built"
"define": "torchtrt_src=prebuilt"
}
)

Expand Down

0 comments on commit 540e135

Please sign in to comment.