@@ -23,66 +23,64 @@ jobs:
23
23
fail-fast : false
24
24
matrix :
25
25
whl :
26
- - url : https://download.pytorch.org/whl/test/cu117_pypi_cudnn/torch-1.13.1%2Bcu117.with.pypi.cudnn-cp310-cp310-linux_x86_64.whl
27
- python : " 3.10" # python version to use for smoke tests
28
- upload_artifact : false # upload the repackaged binary as an artifact
29
- - url : https://download.pytorch.org/whl/test/cu117_pypi_cudnn/torch-1.13.1%2Bcu117.with.pypi.cudnn-cp37-cp37m-linux_x86_64.whl
30
- python : " 3.7"
26
+ - url : https://download.pytorch.org/whl/test/cu121/torch-2.2.0%2Bcu121-cp312-cp312-linux_x86_64.whl
27
+ python : " 3.12"
31
28
artifact : false
32
- - url : https://download.pytorch.org/whl/test/cu117_pypi_cudnn/torch-1.13.1%2Bcu117.with.pypi.cudnn-cp38-cp38-linux_x86_64.whl
33
- python : " 3.8"
29
+ - url : https://download.pytorch.org/whl/test/cu121/torch-2.2.0%2Bcu121-cp311-cp311-linux_x86_64.whl
30
+ python : " 3.11" # python version to use for smoke tests
31
+ upload_artifact : false # upload the repackaged binary as an artifact
32
+ - url : https://download.pytorch.org/whl/test/cu121/torch-2.2.0%2Bcu121-cp310-cp310-linux_x86_64.whl
33
+ python : " 3.10"
34
34
artifact : false
35
- - url : https://download.pytorch.org/whl/test/cu117_pypi_cudnn /torch-1.13.1%2Bcu117.with.pypi.cudnn -cp39-cp39-linux_x86_64.whl
35
+ - url : https://download.pytorch.org/whl/test/cu121 /torch-2.2.0%2Bcu121 -cp39-cp39-linux_x86_64.whl
36
36
python : " 3.9"
37
37
artifact : false
38
- # - url: https://download.pytorch.org/whl/test/cu117_pypi_cudnn /torch-1.13.1%2Bcu117.with.pypi.cudnn-cp311-cp311 -linux_x86_64.whl
39
- # python: "3.11 "
40
- # artifact: false
38
+ - url : https://download.pytorch.org/whl/test/cu121 /torch-2.2.0%2Bcu121-cp38-cp38 -linux_x86_64.whl
39
+ python : " 3.8 "
40
+ artifact : false
41
41
42
42
uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
43
43
with :
44
- runner : linux.4xlarge.nvidia.gpu
44
+ runner : linux.g5. 4xlarge.nvidia.gpu
45
45
job-name : " Validate binary size"
46
46
upload-artifact : ${{ matrix.whl.upload_artifact == 'true' && 'repackaged-binary' || '' }}
47
47
script : |
48
48
set -ex
49
49
export ENV_NAME="conda-env-${{ github.run_id }}"
50
- export GPU_ARCH_VER="11.7 "
51
- export GPU_ARCH_TYPE ="cuda"
52
- export CUDA_VER="11.7 "
50
+ export MATRIX_GPU_ARCH_VERSION="12.1 "
51
+ export MATRIX_GPU_ARCH_TYPE ="cuda"
52
+ export MATRIX_CUDA_VER="12.1 "
53
53
export DESIRED_PYTHON="${{ matrix.whl.python }}"
54
- export DESIRED_CUDA="cu117"
55
- export PACKAGE_TYPE="wheel"
54
+ export MATRIX_PACKAGE_TYPE="wheel"
56
55
export TARGET_OS="linux"
57
- export INSTALLATION=""
58
-
56
+
59
57
# install zip
60
58
sudo yum install zip -y
61
-
59
+
62
60
# install patchelf
63
61
chmod a+x common/install_patchelf.sh
64
62
sudo common/install_patchelf.sh
65
-
63
+
66
64
# download torch whl
67
65
wget ${{ matrix.whl.url }}
68
66
FILENAME=$(ls -1 *.whl | head -n 1)
69
67
SIZE_BEFORE=$(du -h $FILENAME | cut -f1)
70
-
68
+
71
69
# repackage into manywheel
72
70
release/pypi/prep_binary_for_pypi.sh $FILENAME
73
-
71
+
74
72
NEW_FILENAME=$(ls -1 *.whl | head -n 1)
75
73
echo "::notice:: $FILENAME before: $SIZE_BEFORE after: $(du -h $NEW_FILENAME | cut -f1)"
76
-
74
+
77
75
# cp to ${RUNNER_ARTIFACT_DIR}
78
76
cp $NEW_FILENAME ${RUNNER_ARTIFACT_DIR}/
79
-
77
+
80
78
# create conda env
81
79
conda create -y -n $ENV_NAME python=$DESIRED_PYTHON
82
80
conda activate $ENV_NAME
83
-
81
+
84
82
# install torch
85
83
pip install numpy pillow $NEW_FILENAME
86
-
84
+
87
85
# run smoke test
88
- python ./test/smoke_test/smoke_test.py --package=torchonly
86
+ python ./test/smoke_test/smoke_test.py --package=torchonly
0 commit comments