Test check binary pytorch#2

atalman · Sep 29, 2022 · 4fdafaf · 4fdafaf
1 parent 48030d4
commit 4fdafaf
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 2 deletions.
diff --git a/.github/actions/validate-binary/action.yml b/.github/actions/validate-binary/action.yml
@@ -17,6 +17,18 @@ inputs:
     description: 'Python version'
     required: true
     default: '3.9'
+  desired_cuda:
+    description: 'Desired CUDA'
+    required: true
+    default: 'cpu'
+  dev_toolset:
+    description: 'Dev Toolset'
+    required: false
+    default: ''
+  package_type:
+    description: 'Package Type'
+    required: true
+    default: 'conda'
   target_os:
     description: 'Target OS linux or macos'
     required: false
@@ -49,9 +61,16 @@ runs:
         GPU_ARCH_TYPE: ${{ inputs.gpu_arch_type }}
         INSTALLATION: ${{ inputs.installation }}
         ENV_NAME: conda-env-${{ github.run_id }}
+        DESIRED_PYTHON: ${{ inputs.python_version }}
+        DESIRED_CUDA: ${{ inputs.desired_cuda }}
+        DESIRED_DEVTOOLSET: ${{ inputs.dev_toolset }}
+        PACKAGE_TYPE: ${{ inputs.package_type }}
       run: |
         set -ex
         conda create -yp ${ENV_NAME} python=${{ inputs.python_version }} numpy
         conda run -p ${ENV_NAME} $INSTALLATION
         conda run -p ${ENV_NAME} python3  ./test/smoke_test/smoke_test.py
+        export LD_LIBRARY_PATH="$(dirname $(which python))/lib"
+        export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib
+        conda run -p ${ENV_NAME} env LD_LIBRARY_PATH=$LD_LIBRARY_PATH bash ${PWD}/check_binary.sh
         conda env remove -p ${ENV_NAME}
diff --git a/.github/workflows/validate-linux-binaries.yml b/.github/workflows/validate-linux-binaries.yml
@@ -47,13 +47,19 @@ jobs:
       fail-fast: false
     runs-on: ${{ matrix.validation_runner }}
     steps:
+      - name: Checkout PyTorch builder
+        uses: actions/checkout@v2
       - name: Validate binary conda
-        uses: pytorch/builder/.github/actions/validate-binary@main
+        uses: .github/actions/validate-binary
         with:
           gpu_arch_type: ${{ matrix.gpu_arch_type }}
           gpu_arch_ver: ${{ matrix.gpu_arch_version }}
           installation: ${{ matrix.installation }}
           python_version: ${{ matrix.python_version }}
+          desired_cuda: ${{ matrix.desired_cuda }}
+          dev_toolset: ''
+          package_type: conda
+          target_os: linux
   validate-linux-binaries-wheels:
     needs: generate-linux-wheel-matrix
     strategy:
@@ -62,13 +68,19 @@ jobs:
       fail-fast: false
     runs-on: ${{ matrix.validation_runner }}
     steps:
+      - name: Checkout PyTorch builder
+        uses: actions/checkout@v2
       - name: Validate binary wheel
-        uses: pytorch/builder/.github/actions/validate-binary@main
+        uses: .github/actions/validate-binary
         with:
           gpu_arch_type: ${{ matrix.gpu_arch_type }}
           gpu_arch_ver: ${{ matrix.gpu_arch_version }}
           installation: ${{ matrix.installation }}
           python_version: ${{ matrix.python_version }}
+          desired_cuda: ${{ matrix.desired_cuda }}
+          dev_toolset: ''
+          package_type: wheel
+          target_os: linux
   validate-linux-libtorch-binaries:
     needs: generate-linux-libtorch-matrix
     strategy: