diff --git a/docker/dist-test.sh b/docker/dist-test.sh index 519f71634f..a4998460b8 100755 --- a/docker/dist-test.sh +++ b/docker/dist-test.sh @@ -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 diff --git a/tests/README.md b/tests/README.md index 6c7ebc0cac..2d1a9c493a 100644 --- a/tests/README.md +++ b/tests/README.md @@ -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. diff --git a/tests/util/BUILD b/tests/util/BUILD index 385df38b79..f233bd0563 100644 --- a/tests/util/BUILD +++ b/tests/util/BUILD @@ -10,7 +10,7 @@ config_setting( config_setting( name = "ci_build_testing", values = { - "define": "torchtrt_src=pre_built" + "define": "torchtrt_src=prebuilt" } )