test

Author: lanluo-nvidia

.github/workflows/build-tensorrt-linux.yml

@@ -70,6 +70,10 @@ on:
         required: false
         type: boolean
         default: true
+      upload-artifact:
+        description: 'Name to give artifacts uploaded from ${inputs.repository}/dist'
+        default: ''
+        type: string
 
 permissions:
   id-token: write
@@ -213,7 +217,7 @@ jobs:
         continue-on-error: true
         uses: actions/upload-artifact@v3
         with:
-          name: torch_tensorrt_${{env.TENSORRT_VERSION}}_py${{env.PYTHON_VERSION}}_${{env.CU_VERSION}}
+          name: ${{ inputs.upload-artifact }}
           path: ${{ inputs.repository }}/dist
 
 concurrency:

.github/workflows/build-test-tensorrt-linux.yml

@@ -67,6 +67,7 @@ jobs:
       package-name: ${{ matrix.package-name }}
       smoke-test-script: ${{ matrix.smoke-test-script }}
       trigger-event: ${{ github.event_name }}
+      upload-artifact: torch_tensorrt_${{ matrix.tensorrt.version }}_py${{ matrix.python_version }}_${{ matrix.desired_cuda }}
 
   tests-py-torchscript-fe:
     name: Test torchscript frontend [Python]
@@ -89,6 +90,7 @@ jobs:
       test-infra-ref: main
       build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
+      download-artifact: torch_tensorrt_${{ matrix.tensorrt.version }}_py${{ matrix.python_version }}_${{ matrix.desired_cuda }}
       script: |
         export USE_HOST_DEPS=1
         export CI_BUILD=1

.github/workflows/linux-test.yml

@@ -49,6 +49,10 @@ on:
         description: "Prevents a job from failing when a step fails. Set to true to allow a job to pass when exec script step fails."
         default: false
         type: boolean
+      download-artifact:
+        description: 'Name of the artifacts to be downloaded'
+        default: ''
+        type: string
       upload-artifact:
         description: 'Name to give artifacts uploaded from ${RUNNER_ARTIFACT_DIR}'
         default: ''
@@ -112,10 +116,17 @@ jobs:
           repository: ${{ inputs.repository  }}
           script: ${{ inputs.pre-script }}
       - name: Download artifacts
+        if: ${{ inputs.download-artifact == '' }}
         uses: actions/download-artifact@v3
         with:
           name: ${{ env.ARTIFACT_NAME }}
           path: /opt/torch-tensorrt-builds/
+      - name: Download artifacts
+        if: ${{ inputs.download-artifact != '' }}
+        uses: actions/download-artifact@v3
+        with:
+          name: ${{ inputs.download-artifact }}
+          path: /opt/torch-tensorrt-builds/
       # - name: Install torch and torch-tensorrt
       #   if: ${{ inputs.pre-script != '' }}
       #   uses: ./test-infra/.github/actions/run-script-with-cache