diff --git a/.github/workflows/linux-release-artifacts.yml b/.github/workflows/linux-release-artifacts.yml index d295387ab0..d755fd16aa 100644 --- a/.github/workflows/linux-release-artifacts.yml +++ b/.github/workflows/linux-release-artifacts.yml @@ -41,6 +41,10 @@ on: required: false type: boolean default: true + channel: + description: "Channel to use (nightly, test, release, all)" + default: "" + type: string # script: # description: 'Script to utilize' # default: "python setup.py bdist_wheel" @@ -73,7 +77,7 @@ jobs: container: image: ${{ matrix.container_image }} options: ${{ matrix.gpu_arch_type == 'cuda' && '--gpus all' || ' ' }} - if: ${{ env.CU_VERSION == '12.1' }} + if: ${{ matrix.desired_cuda == '12.1' }} # If a build is taking longer than 60 minutes on these runners we need # to have a conversation timeout-minutes: 60 diff --git a/.github/workflows/linux-release.yml b/.github/workflows/linux-release.yml index e7dabc1b87..ea6849e704 100644 --- a/.github/workflows/linux-release.yml +++ b/.github/workflows/linux-release.yml @@ -48,7 +48,7 @@ jobs: smoke-test-script: ${{ matrix.smoke-test-script }} trigger-event: ${{ github.event_name }} - build-other-release-artifacts: + build-release-artifacts: name: Build other release artifacts needs: [generate-matrix, build] strategy: @@ -56,10 +56,10 @@ jobs: matrix: include: - repository: pytorch/tensorrt - if: ${{ env.CHANNEL == 'release'}} uses: ./.github/workflows/linux-release-artifacts.yml + if: ${{ contains( github.event.pull_request.labels.*.name, 'build-release-artifacts') }} with: - job-name: build-other-release-artifiacts + job-name: build-release-artifiacts repository: "pytorch/tensorrt" ref: "" test-infra-repository: pytorch/test-infra