install onnxruntime tests on linux wheel for circle_ci

packaging/build_conda.sh

@@ -11,4 +11,15 @@ setup_conda_pytorch_constraint
 setup_conda_cudatoolkit_constraint
 setup_visual_studio_constraint
 setup_junit_results_folder
+
+# Small workaround to install onnxruntime on the machine.
+# just to see if it works before cleanup
+if [[ "$(uname)" == Linux ]] ; then
+    conda create -n build_env python="$PYTHON_VERSION" anaconda -yq
+    activate build_env
+    conda install -yq -c anaconda protobuf
+    conda install -yq -c conda-forge pybind11
+    "$(conda info --base)/envs/build_env/bin/python" -m pip -q install onnxruntime
+fi
+
 conda build $CONDA_CHANNEL_FLAGS -c defaults -c conda-forge --no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchvision

packaging/build_wheel.sh

@@ -10,4 +10,5 @@ setup_wheel_python
 pip_install numpy pyyaml future ninja
 setup_pip_pytorch_version
 python setup.py clean
+install_onnx_runtime_on_linux_wheel
 IS_WHEEL=1 python setup.py bdist_wheel

packaging/pkg_helpers.bash

@@ -276,3 +276,27 @@ setup_junit_results_folder() {
     export CONDA_PYTORCH_BUILD_RESULTS_DIRECTORY="${SOURCE_ROOT_DIR}/build_results/results.xml"
   fi
 }
+
+install_onnx_runtime_on_linux_wheel() { 
+  if [[ "$(uname)" == Linux ]] ; then
+    case "$PYTHON_VERSION" in
+      # TODO: Enable when onnxruntime is packaged for python 3.8
+      3.8) ;;
+      *)
+        pip_install onnxruntime 
+        ;;
+    esac
+  fi
+}
+
+install_onnx_runtime_on_linux_conda() { 
+  if [[ "$(uname)" == Linux ]] ; then
+    case "$PYTHON_VERSION" in
+      # TODO: Enable when onnxruntime is packaged for python 3.8
+      3.8) ;;
+      *)
+        "$(conda info --base)/bin/python" -m pip -q install onnxruntime
+        ;;
+    esac
+  fi
+}