llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use delocate to make portable wheels on OSX (#850) 

Fix up wheels per python version on OSX
pull/852/head snapshot-20220512.447
powderluv 2022-05-12 14:16:32 -07:00 committed by GitHub
parent ec0e9e0bc7
commit e7f306ec2f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 31 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
build_tools/python_deploy/build_macos_packages.sh
View File

@ -20,7 +20,7 @@ set -eu -o errtrace
this_dir="$(cd $(dirname $0) && pwd)" this_dir="$(cd $(dirname $0) && pwd)"
repo_root="$(cd $this_dir/../../ && pwd)" repo_root="$(cd $this_dir/../../ && pwd)"
python_versions="${python_versions:-3.9 3.10}" python_versions="${TORCH_MLIR_PYTHON_VERSIONS:-3.9 3.10}"
output_dir="${output_dir:-${this_dir}/wheelhouse}" output_dir="${output_dir:-${this_dir}/wheelhouse}"
packages="${packages:-torch-mlir}" packages="${packages:-torch-mlir}"
@ -55,6 +55,7 @@ function run() {
torch-mlir) torch-mlir)
clean_wheels torch-mlir $python_version clean_wheels torch-mlir $python_version
build_torch_mlir build_torch_mlir
run_audit_wheel torch_mlir $python_version
;; ;;
*) *)
echo "Unrecognized package '$package'" echo "Unrecognized package '$package'"
@ -78,7 +79,28 @@ function clean_wheels() {
local wheel_basename="$1" local wheel_basename="$1"
local python_version="$2" local python_version="$2"
echo ":::: Clean wheels $wheel_basename $python_version" echo ":::: Clean wheels $wheel_basename $python_version"
rm -f $output_dir/${wheel_basename}-*-${python_version}-*.whl rm -rf $repo_root/build/
rm -f $output_dir/${wheel_basename}-*-${python_version//./}-*.whl
}
function run_audit_wheel() {
set +x
local wheel_basename="$1"
local python_version="$2"
generic_wheel=`ls $output_dir/${wheel_basename}-* | grep ${python_version//./}`
echo "Looking for $generic_wheel"
if [ -f "$generic_wheel" ]; then
echo "$generic_wheel found. Delocating it.."
rm -rf $output_dir/test_venv
python${python_version} -m venv $output_dir/test_venv
source $output_dir/test_venv/bin/activate
python${python_version} -m pip install -U pip
python${python_version} -m pip install -r $repo_root/requirements.txt --extra-index-url https://download.pytorch.org/whl/nightly/cpu
python${python_version} -m pip install $generic_wheel --extra-index-url https://download.pytorch.org/whl/nightly/cpu
DYLD_LIBRARY_PATH=$output_dir/test_venv/lib/python${python_version}/site-packages/torch/lib delocate-wheel -v $generic_wheel
deactivate
rm -rf $output_dir/test_venv
fi
} }
run run

8
build_tools/python_deploy/build_upload_m1_snapshot.sh
View File

@ -26,9 +26,15 @@ package_version=${latest_snapshot#"snapshot-"}
echo "Latest snapshot tag is: ${latest_snapshot}" echo "Latest snapshot tag is: ${latest_snapshot}"
echo "Latest version is: ${package_version}" echo "Latest version is: ${package_version}"
export TORCH_MLIR_PYTHON_VERSIONS="3.9 3.10"
echo "Using Python Versions: ${TORCH_MLIR_PYTHON_VERSIONS}"
export TORCH_MLIR_PYTHON_PACKAGE_VERSION="${package_version}" export TORCH_MLIR_PYTHON_PACKAGE_VERSION="${package_version}"
echo "Setting torch-mlir Python Package version to: ${TORCH_MLIR_PYTHON_PACKAGE_VERSION}" echo "Setting torch-mlir Python Package version to: ${TORCH_MLIR_PYTHON_PACKAGE_VERSION}"
TORCH_MLIR_PYTHON_PACKAGE_VERSION="${package_version}" ${repo_root}/build_tools/python_deploy/build_macos_packages.sh TORCH_MLIR_OSX_ARCH=arm64 \
TORCH_MLIR_OSX_TARGET=11.0 \
TORCH_MLIR_PYTHON_PACKAGE_VERSION="${package_version}" \
TORCH_MLIR_PYTHON_VERSIONS="${TORCH_MLIR_PYTHON_VERSIONS}" \
${repo_root}/build_tools/python_deploy/build_macos_packages.sh
gh release upload ${latest_snapshot} ${repo_root}/build_tools/python_deploy/wheelhouse/torch*.whl gh release upload ${latest_snapshot} ${repo_root}/build_tools/python_deploy/wheelhouse/torch*.whl