CHANNELS

Author: gs-olive

.github/scripts/install-torch-tensorrt.sh

@@ -2,8 +2,7 @@
 set -eou pipefail
 # Source conda so it's available to the script environment
 source ${BUILD_ENV_FILE}
-${CONDA_RUN} python -m pip install pyyaml
-# ${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision pyyaml
+${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision pyyaml
 export TRT_VERSION=$(${CONDA_RUN} python -c "import versions; versions.tensorrt_version()")
 ${CONDA_RUN} python -m pip install /opt/torch-tensorrt-builds/torch_tensorrt*+${CU_VERSION}*.whl tensorrt~=${TRT_VERSION} tensorrt-bindings~=${TRT_VERSION} --extra-index-url=https://pypi.ngc.nvidia.com
 

.github/workflows/build-test.yml

@@ -13,6 +13,9 @@ on:
         - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
   workflow_dispatch:
 
+env:
+  channel: test
+
 jobs:
   generate-matrix:
     uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@release/2.1
@@ -21,7 +24,7 @@ jobs:
       os: linux
       test-infra-repository: pytorch/test-infra
       test-infra-ref: release/2.1
-      channel: test
+      channel: ${{ env.channel }}
       with-rocm: false
       with-cpu: false
 
@@ -109,15 +112,13 @@ jobs:
       test-infra-ref: release/2.1
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
+      channel: ${{ env.channel }}
       script: |
         export USE_HOST_DEPS=1
-        export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
         pushd .
         cd tests/py/dynamo
-        ${CONDA_RUN} python -m pip install --pre pytest timm transformers parameterized expecttest --use-deprecated=legacy-resolver
-        ${CONDA_RUN} python -c "import torch; print('IMPORTED TORCH')"
-        ${CONDA_RUN} python -c "import tensorrt; print('IMPORTED TRT')"
-        ${CONDA_RUN} python -c "import torch_tensorrt; torch_tensorrt.dump_build_info()"
+        ${CONDA_RUN} python -m pip install --pre pytest-xdist timm transformers parameterized expecttest --use-deprecated=legacy-resolver
+        ${CONDA_RUN} python -m pytest --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 10 conversion/
         popd
 
   # tests-py-dynamo-fe:

.github/workflows/linux-test.yml

@@ -53,6 +53,10 @@ on:
         description: 'Name to give artifacts uploaded from ${RUNNER_ARTIFACT_DIR}'
         default: ''
         type: string
+      channel:
+        description: "Channel to use (nightly, test, release, all)"
+        default: ""
+        type: string
 
 jobs:
   test:
@@ -66,6 +70,7 @@ jobs:
       REF: ${{ inputs.ref }}
       CU_VERSION: ${{ matrix.desired_cuda }}
       SCRIPT: ${{ inputs.script }}
+      CHANNEL: ${{ inputs.channel }}
       RUNNER_TEST_RESULTS_DIR: /tmp/test_results
     name: ${{ inputs.job-name }}-${{ matrix.desired_cuda }}
     runs-on: ${{ matrix.validation_runner }}