test

pytorch · Jan 24, 2024 · a906f12 · a906f12
1 parent 89745ca
commit a906f12
Showing 1 changed file with 21 additions and 21 deletions.
diff --git a/.github/workflows/validate-repackaged-binary-sizes.yml b/.github/workflows/validate-repackaged-binary-sizes.yml
@@ -23,21 +23,21 @@ jobs:
       fail-fast: false
       matrix:
         whl:
-          - url: https://download.pytorch.org/whl/test/cu117_pypi_cudnn/torch-1.13.1%2Bcu117.with.pypi.cudnn-cp310-cp310-linux_x86_64.whl
-            python: "3.10"  # python version to use for smoke tests
-            upload_artifact: false # upload the repackaged binary as an artifact
-          - url: https://download.pytorch.org/whl/test/cu117_pypi_cudnn/torch-1.13.1%2Bcu117.with.pypi.cudnn-cp37-cp37m-linux_x86_64.whl
-            python: "3.7"
+          - url: https://download.pytorch.org/whl/test/cu121/torch-2.2.0-cu121-cp312-cp312-linux_x86_64.whl
+            python: "3.12"
             artifact: false
-          - url: https://download.pytorch.org/whl/test/cu117_pypi_cudnn/torch-1.13.1%2Bcu117.with.pypi.cudnn-cp38-cp38-linux_x86_64.whl
-            python: "3.8"
+          - url: https://download.pytorch.org/whl/test/cu121/torch-2.2.0-cu121-cp311-cp311-linux_x86_64.whl
+            python: "3.11"  # python version to use for smoke tests
+            upload_artifact: false # upload the repackaged binary as an artifact
+          - url: https://download.pytorch.org/whl/test/cu121/torch-2.2.0-cu121-cp310-cp310-linux_x86_64.whl
+            python: "3.10"
             artifact: false
-          - url: https://download.pytorch.org/whl/test/cu117_pypi_cudnn/torch-1.13.1%2Bcu117.with.pypi.cudnn-cp39-cp39-linux_x86_64.whl
+          - url: https://download.pytorch.org/whl/test/cu121/torch-2.2.0-cu121-cp319-cp319-linux_x86_64.whl
             python: "3.9"
             artifact: false
-     #    - url: https://download.pytorch.org/whl/test/cu117_pypi_cudnn/torch-1.13.1%2Bcu117.with.pypi.cudnn-cp311-cp311-linux_x86_64.whl
-     #      python: "3.11"
-     #      artifact: false
+          - url: https://download.pytorch.org/whl/test/cu121/torch-2.2.0-cu121-cp318-cp318-linux_x86_64.whl
+            python: "3.8"
+            artifact: false
 
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
@@ -55,34 +55,34 @@ jobs:
         export PACKAGE_TYPE="wheel"
         export TARGET_OS="linux"
         export INSTALLATION=""
-        
+
         # install zip
         sudo yum install zip -y
-        
+
         # install patchelf
         chmod a+x common/install_patchelf.sh
         sudo common/install_patchelf.sh
-        
+
         # download torch whl
         wget ${{ matrix.whl.url }}
         FILENAME=$(ls -1 *.whl | head -n 1)
         SIZE_BEFORE=$(du -h $FILENAME | cut -f1)
-        
+
         # repackage into manywheel
         release/pypi/prep_binary_for_pypi.sh $FILENAME
-        
+
         NEW_FILENAME=$(ls -1 *.whl | head -n 1)
         echo "::notice:: $FILENAME before: $SIZE_BEFORE after: $(du -h $NEW_FILENAME | cut -f1)"
-        
+
         # cp to ${RUNNER_ARTIFACT_DIR}
         cp $NEW_FILENAME ${RUNNER_ARTIFACT_DIR}/
-        
+
         # create conda env
         conda create -y -n $ENV_NAME python=$DESIRED_PYTHON
         conda activate $ENV_NAME
-        
+
         # install torch
         pip install numpy pillow $NEW_FILENAME
-        
+
         # run smoke test
-        python ./test/smoke_test/smoke_test.py --package=torchonly
+        python ./test/smoke_test/smoke_test.py --package=torchonly