|
50 | 50 | default: false |
51 | 51 | type: boolean |
52 | 52 | upload-artifact: |
53 | | - description: 'Name to give wheel artifacts uploaded from the repository dist directory' |
| 53 | + description: 'Name prefix for wheel artifacts uploaded from the repository dist directory. This workflow appends -<python_version>-<desired_cuda>-<architecture>, plus -python-only and -rtx when those inputs are set. Rows that differ only in matrix.tensorrt.version are not separated by it.' |
54 | 54 | default: '' |
55 | 55 | type: string |
56 | 56 | additional-artifact: |
57 | | - description: "Optional additional artifact to install alongside the core wheel" |
| 57 | + description: "Name prefix of an optional additional artifact to install alongside the core wheel. The same suffix is appended, so it lines up with what upload-artifact produced when both calls are given the same build matrix and the same architecture." |
58 | 58 | default: "" |
59 | 59 | type: string |
60 | 60 | use-rtx: |
|
94 | 94 | ARCH: ${{ inputs.architecture }} |
95 | 95 | USE_TRT_RTX: ${{ inputs.use-rtx }} |
96 | 96 | DOWNLOAD_ARTIFACT_NAME: pytorch_tensorrt_${{ matrix.tensorrt.version }}_${{ matrix.python_version }}_${{ matrix.desired_cuda }}_${{ inputs.architecture }}${{ inputs.python-only && '_python_only' || '' }}${{ inputs.use-rtx && '_rtx' || '' }} |
| 97 | + # matrix.* and inputs.architecture are out of scope at a workflow_call site, so a |
| 98 | + # per-row suffix can only be built here. The stem stays with the caller, and the |
| 99 | + # producer and the consumer have to pass the same one. |
| 100 | + ARTIFACT_NAME_SUFFIX: ${{ matrix.python_version }}-${{ matrix.desired_cuda }}-${{ inputs.architecture }}${{ inputs.python-only && '-python-only' || '' }}${{ inputs.use-rtx && '-rtx' || '' }} |
97 | 101 | name: ${{ inputs.job-name }}-${{ matrix.tensorrt.version }}-${{ matrix.python_version }}-${{ matrix.desired_cuda }} |
98 | 102 | runs-on: ${{ inputs.runner != '' && inputs.runner || matrix.validation_runner }} |
99 | 103 | container: |
@@ -168,7 +172,7 @@ jobs: |
168 | 172 | if: ${{ inputs.additional-artifact != '' }} |
169 | 173 | uses: actions/download-artifact@v7 |
170 | 174 | with: |
171 | | - name: ${{ inputs.additional-artifact }} |
| 175 | + name: ${{ inputs.additional-artifact }}-${{ env.ARTIFACT_NAME_SUFFIX }} |
172 | 176 | path: /opt/torch-tensorrt-builds/ |
173 | 177 | - name: Pack script |
174 | 178 | continue-on-error: ${{ inputs.continue-on-error }} |
@@ -226,7 +230,7 @@ jobs: |
226 | 230 | uses: actions/upload-artifact@v6 |
227 | 231 | if: ${{ inputs.upload-artifact != '' }} |
228 | 232 | with: |
229 | | - name: ${{ inputs.upload-artifact }} |
| 233 | + name: ${{ inputs.upload-artifact }}-${{ env.ARTIFACT_NAME_SUFFIX }} |
230 | 234 | path: ${{ inputs.repository }}/dist/ |
231 | 235 | if-no-files-found: error |
232 | 236 |
|
|
0 commit comments