Build script cleanup

Author: borisfom

.bazelrc

@@ -33,4 +33,10 @@ build:python --define=target_lang=python
 
 build:pre_cxx11_abi --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"
 build:pre_cxx11_abi --linkopt="-D_GLIBCXX_USE_CXX11_ABI=0"
-build:pre_cxx11_abi --define=abi=pre_cxx11_abi
+build:pre_cxx11_abi --define=abi=pre_cxx11_abi
+
+build:ci_testing --define=torchtrt_src=pre_built
+build:use_precompiled_torchtrt --define=torchtrt_src=pre_built
+
+test:ci_testing --define=torchtrt_src=pre_built
+test:use_precompiled_torchtrt --define=torchtrt_src=pre_built

docker/dist-build.sh

@@ -1,8 +1,13 @@
 #!/bin/bash
 
-mkdir -p dist && cd py \
-    && MAX_JOBS=1 LANG=en_US.UTF-8 LANGUAGE=en_US:en LC_ALL=en_US.UTF-8 python3 setup.py bdist_wheel --use-cxx11-abi \
-    && cp -r dist/* ../dist/
+TOP_DIR=$(cd $(dirname $0); pwd)/..
+
+cd ${TOP_DIR} \
+    && mkdir -p dist && cd py \
+    && MAX_JOBS=1 LANG=en_US.UTF-8 LANGUAGE=en_US:en LC_ALL=en_US.UTF-8 \
+	       python3 setup.py bdist_wheel --use-cxx11-abi $* || exit 1
+								
+cd ${TOP_DIR} && cp -r dist/* ../dist/
 
 pip3 install ipywidgets --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org
 jupyter nbextension enable --py widgetsnbextension

docker/dist-test.sh

@@ -5,4 +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
+bazel test //tests:tests //tests:python_api_tests --compilation_mode=opt --jobs=4 --define=torchtrt_src=prebuilt
+