test

pytorch · Nov 5, 2024 · 7102fa5 · 7102fa5
1 parent a554285
commit 7102fa5
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 11 deletions.
diff --git a/.github/scripts/generate-tensorrt-test-matrix.py b/.github/scripts/generate-tensorrt-test-matrix.py
@@ -66,7 +66,7 @@ def main(args: list[str]) -> None:
 
     options = parser.parse_args(args)
     if options.matrix == "":
-        raise Exception(f"--matrix is empty, please provide the matrix json str")
+        raise Exception("--matrix is empty, please provide the matrix json str")
 
     matrix_dict = json.loads(options.matrix)
     includes = matrix_dict["include"]

diff --git a/.github/workflows/build-tensorrt-linux.yml b/.github/workflows/build-tensorrt-linux.yml
@@ -92,7 +92,7 @@ jobs:
       TENSORRT_VERSION: ${{ matrix.tensorrt.version }}
       TENSORRT_URLS: ${{ matrix.tensorrt.urls }}
       TENSORRT_SHA256: ${{ matrix.tensorrt.sha256 }}
-      ARTIFACT_NAME: torch_tensorrt_${{ matrix.tensorrt.version }}_py${{ matrix.python_version }}_${{ matrix.desired_cuda }}
+      UPLOAD_ARTIFACT_NAME: torch_tensorrt_${{ matrix.tensorrt.version }}_py${{ matrix.python_version }}_${{ matrix.desired_cuda }}
     name: build_tensorrt${{ matrix.tensorrt.version }}_py${{matrix.python_version}}_${{matrix.desired_cuda}}
     runs-on: ${{ matrix.validation_runner }}
     container:
@@ -214,7 +214,7 @@ jobs:
         continue-on-error: true
         uses: actions/upload-artifact@v3
         with:
-          name: ${{ env.ARTIFACT_NAME }}
+          name: ${{ env.UPLOAD_ARTIFACT_NAME }}
           path: ${{ inputs.repository }}/dist
 
 concurrency:

diff --git a/.github/workflows/build-test-tensorrt-linux.yml b/.github/workflows/build-test-tensorrt-linux.yml
@@ -89,7 +89,6 @@ jobs:
       test-infra-ref: main
       build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
-      download-artifact: ${{ needs.generate-tensorrt-matrix.outputs.matrix.tensorrt.version }}_py${{ needs.generate-tensorrt-matrix.outputs.matrix.python_version }}_${{ needs.generate-tensorrt-matrix.outputs.matrix.desired_cuda }}
       script: |
         export USE_HOST_DEPS=1
         export CI_BUILD=1

diff --git a/.github/workflows/linux-test.yml b/.github/workflows/linux-test.yml
@@ -49,10 +49,6 @@ 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: ''
@@ -72,6 +68,7 @@ jobs:
       SCRIPT: ${{ inputs.script }}
       RUNNER_TEST_RESULTS_DIR: /tmp/test_results
       ARCH: ${{ inputs.architecture }}
+      DOWNLOAD_ARTIFACT_NAME: torch_tensorrt_${{ matrix.tensorrt.version }}_py${{ matrix.python_version }}_${{ matrix.desired_cuda }}
     name: ${{ inputs.job-name }}-${{ matrix.desired_cuda }}
     runs-on: ${{ matrix.validation_runner }}
     container:
@@ -116,16 +113,16 @@ jobs:
           repository: ${{ inputs.repository  }}
           script: ${{ inputs.pre-script }}
       - name: Download artifacts
-        if: ${{ inputs.download-artifact == '' }}
+        if: ${{ matrix.tensorrt == '' }}
         uses: actions/download-artifact@v3
         with:
           name: ${{ env.ARTIFACT_NAME }}
           path: /opt/torch-tensorrt-builds/
       - name: Download artifacts
-        if: ${{ inputs.download-artifact != '' }}
+        if: ${{ matrix.tensorrt != '' }}
         uses: actions/download-artifact@v3
         with:
-          name: ${{ inputs.download-artifact }}
+          name: ${{ env.DOWNLOAD_ARTIFACT_NAME }}
           path: /opt/torch-tensorrt-builds/
       # - name: Install torch and torch-tensorrt
       #   if: ${{ inputs.pre-script != '' }}