|
59 | 59 | export PYTORCH_BUILD_VERSION=$build_version
|
60 | 60 | export PYTORCH_BUILD_NUMBER=$build_number
|
61 | 61 |
|
| 62 | +package_type="${PACKAGE_TYPE:-wheel}" |
62 | 63 | # Fill in empty parameters with defaults
|
63 | 64 | if [[ -z "$TORCH_PACKAGE_NAME" ]]; then
|
64 | 65 | TORCH_PACKAGE_NAME='torch'
|
@@ -194,15 +195,17 @@ python setup.py bdist_wheel -d "$whl_tmp_dir"
|
194 | 195 |
|
195 | 196 | echo "Finished setup.py bdist_wheel at $(date)"
|
196 | 197 |
|
197 |
| -echo "delocating wheel dependencies" |
198 |
| -retry pip install https://github.com/matthew-brett/delocate/archive/master.zip |
199 |
| -echo "found the following wheels:" |
200 |
| -find $whl_tmp_dir -name "*.whl" |
201 |
| -echo "running delocate" |
202 |
| -find $whl_tmp_dir -name "*.whl" | xargs -I {} delocate-wheel -v {} |
203 |
| -find $whl_tmp_dir -name "*.whl" |
204 |
| -find $whl_tmp_dir -name "*.whl" | xargs -I {} delocate-listdeps {} |
205 |
| -echo "Finished delocating wheels at $(date)" |
| 198 | +if [[ $package_type != 'libtorch' ]]; then |
| 199 | + echo "delocating wheel dependencies" |
| 200 | + retry pip install https://github.com/matthew-brett/delocate/archive/master.zip |
| 201 | + echo "found the following wheels:" |
| 202 | + find $whl_tmp_dir -name "*.whl" |
| 203 | + echo "running delocate" |
| 204 | + find $whl_tmp_dir -name "*.whl" | xargs -I {} delocate-wheel -v {} |
| 205 | + find $whl_tmp_dir -name "*.whl" |
| 206 | + find $whl_tmp_dir -name "*.whl" | xargs -I {} delocate-listdeps {} |
| 207 | + echo "Finished delocating wheels at $(date)" |
| 208 | +fi |
206 | 209 |
|
207 | 210 | echo "The wheel is in $(find $whl_tmp_dir -name '*.whl')"
|
208 | 211 |
|
|
0 commit comments