test

pytorch · Jun 17, 2024 · 6b1b7d2 · 6b1b7d2
1 parent 9ca6c2a
commit 6b1b7d2
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 10 deletions.
diff --git a/.github/scripts/build-linux-release-artifacts.sh b/.github/scripts/build-linux-release-artifacts.sh
@@ -6,13 +6,20 @@ export CXX=g++
 CURRENT_DIR=`pwd`
 
 if [[ "${PYTHON_VERSION}" == "3.8" ]]; then
+  PY_SINGLE_BUILD_CODE=cp38
   PY_BUILD_CODE=cp38-cp38
 elif [[ "${PYTHON_VERSION}" == "3.9" ]]; then
+  PY_SINGLE_BUILD_CODE=cp39
   PY_BUILD_CODE=cp39-cp39
 elif [[ "${PYTHON_VERSION}" == "3.10" ]]; then
+  PY_SINGLE_BUILD_CODE=cp310
   PY_BUILD_CODE=cp310-cp310
 elif [[ "${PYTHON_VERSION}" == "3.11" ]]; then
-  PY_BUILD_CODE=cp38-cp38
+  PY_SINGLE_BUILD_CODE=cp311
+  PY_BUILD_CODE=cp311-cp311
+elif [[ "${PYTHON_VERSION}" == "3.12" ]]; then
+  PY_SINGLE_BUILD_CODE=cp312
+  PY_BUILD_CODE=cp312-cp312
 else
    echo "python version: ${PYTHON_VERSION} is not supported"
    exit
@@ -33,7 +40,9 @@ wget -q https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt
 && tar -xvf TensorRT-10.0.1.6.Linux.x86_64-gnu.cuda-12.4.tar \
 && rm TensorRT-10.0.1.6.Linux.x86_64-gnu.cuda-12.4.tar
 
-TENSERRT_DIR=${CURRENT_DIR}/TensorRT-10.0.1.6
+TENSERRT_DIR=${CURRENT_DIR}/TensorRT-10.0.1.6/
+
+python -m pip install ${TENSERRT_DIR}/python/tensorrt-${TRT_VERSION}-${PY_SINGLE_BUILD_CODE}-none-linux_x86_64.whl
 
 SITE_PKG_DIR=`python -c  'import sysconfig; print(sysconfig.get_paths()["purelib"])'`
 

diff --git a/.github/workflows/linux-release-artifacts.yml b/.github/workflows/linux-release-artifacts.yml
@@ -77,7 +77,9 @@ jobs:
     container:
       image: ${{ matrix.container_image }}
       options: ${{ matrix.gpu_arch_type == 'cuda' && '--gpus all' || ' ' }}
+    # TO DO: ask Naren, I am not able to skip the release build if it is not cuda12.1 on the job level
     # if: ${{ env.CU_VERSION == '12.1' }}
+
     # If a build is taking longer than 60 minutes on these runners we need
     # to have a conversation
     timeout-minutes: 60
@@ -127,12 +129,12 @@ jobs:
         with:
           name: ${{ env.ARTIFACT_NAME }}
           path: /opt/torch-tensorrt-builds/
-      # - name: Install torch and torch-tensorrt
-      #   if: ${{ inputs.pre-script != '' }}
-      #   uses: ./test-infra/.github/actions/run-script-with-cache
-      #   with:
-      #     repository: ${{ inputs.repository  }}
-      #     script: .github/scripts/install-torch-tensorrt.sh
+      - name: Install torch and torch-tensorrt
+        if: ${{ inputs.pre-script != '' }}
+        uses: ./test-infra/.github/actions/run-script-with-cache
+        with:
+          repository: ${{ inputs.repository  }}
+          script: .github/scripts/install-torch-tensorrt.sh
       - name: Pack script
         continue-on-error: ${{ inputs.continue-on-error }}
         working-directory: ${{ inputs.repository }}
@@ -142,7 +144,6 @@ jobs:
           set -euxo pipefail
           # shellcheck disable=SC2086
           source "${BUILD_ENV_FILE}"
-          echo "lan added"
           {
             echo "${SCRIPT}";
           } > "user_script"

diff --git a/py/ci/soname_excludes.params b/py/ci/soname_excludes.params
@@ -42,4 +42,6 @@
 --exclude libnvinfer_plugin.so.10
 --exclude libnvinfer-ea523553.so.10.0.1
 --exclude libcudart-9335f6a2.so.12.1.105
---exclude libnvinfer_plugin-24528888.so.10.0.1
+--exclude libnvinfer_plugin-24528888.so.10.0.1
+
+--exclude libcudart-d0da41ae.so.11.8.89