test

pytorch · Jun 17, 2024 · da36c13 · da36c13
1 parent 4e34a62
commit da36c13
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/.github/workflows/linux-release-artifacts.yml b/.github/workflows/linux-release-artifacts.yml
@@ -45,10 +45,10 @@ on:
         description: "Channel to use (nightly, test, release, all)"
         default: ""
         type: string
-      # script:
-      #   description: 'Script to utilize'
-      #   default: "python setup.py bdist_wheel"
-      #   type: string
+      script:
+        description: 'Script to utilize'
+        default: "python setup.py bdist_wheel"
+        type: string
       continue-on-error:
         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
@@ -69,10 +69,10 @@ jobs:
       REPOSITORY: ${{ inputs.repository }}
       REF: ${{ inputs.ref }}
       CU_VERSION: ${{ matrix.desired_cuda }}
-      # SCRIPT: ${{ inputs.script }}
-      # RUNNER_TEST_RESULTS_DIR: /tmp/test_results
+      SCRIPT: ${{ inputs.script }}
+      RUNNER_TEST_RESULTS_DIR: /tmp/test_results
       ARCH: ${{ inputs.architecture }}
-    name: ${{ inputs.job-name }}-${{ matrix.desired_cuda }}
+    name: ${{ inputs.job-name }}-${{ matrix.python_version }}-${{ matrix.desired_cuda }}
     runs-on: ${{ matrix.validation_runner }}
     container:
       image: ${{ matrix.container_image }}
@@ -134,7 +134,9 @@ jobs:
           ALL_SECRETS: ${{ toJSON(secrets) }}
         run: |
           set -euxo pipefail
+          # shellcheck disable=SC2086
           source "${BUILD_ENV_FILE}"
+          echo "lan added"
           {
             echo "${SCRIPT}";
           } > "user_script"

diff --git a/.github/workflows/linux-release.yml b/.github/workflows/linux-release.yml
@@ -55,7 +55,7 @@ jobs:
           - repository: pytorch/tensorrt
           - package-name: torch_tensorrt
     uses: ./.github/workflows/linux-release-artifacts.yml
-    if: ${{ contains( github.event.pull_request.labels.*.name, 'build-release-artifacts') }}
+    if: ${{ contains(github.event.pull_request.labels.*.name, 'build-release-artifacts') }}
     with:
       job-name: build-release-artifiacts
       repository: "pytorch/tensorrt"
@@ -64,6 +64,8 @@ jobs:
       test-infra-ref: main
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
+      script: |
+        export USE_HOST_DEPS=1
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref_name }}-${{ inputs.repository }}-${{ inputs.job-name }}