Skip to content

Commit

Permalink
feat: Updating the pre_built to prebuilt
Browse files Browse the repository at this point in the history
Signed-off-by: Anurag Dixit <[email protected]>
  • Loading branch information
Anurag Dixit committed Nov 2, 2021
1 parent 5e0880b commit 51412c7
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 51412c7

Please sign in to comment.