Skip to content

Commit 67b189d

Browse files
exclude torch binaries from whl (#267)
1 parent db72fb8 commit 67b189d

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

ci/wheel/build_whl.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ build_audit_whl() {
1212
/opt/python/$1/bin/python setup.py clean bdist_wheel
1313

1414
/opt/python/$1/bin/python -m pip install auditwheel
15-
LD_LIBRARY_PATH=${TORCH_LIB_DIR}:${LD_LIBRARY_PATH} /opt/python/$1/bin/python -m auditwheel repair --plat manylinux_2_28_x86_64 dist/torch_migraphx-*-$1-linux_x86_64.whl
15+
LD_LIBRARY_PATH=${TORCH_LIB_DIR}:${LD_LIBRARY_PATH} /opt/python/$1/bin/python -m auditwheel repair $(cat ${PROJECT_DIR}/ci/wheel/excludes.params) --plat manylinux_2_28_x86_64 dist/torch_migraphx-*-$1-linux_x86_64.whl
1616
}
1717

1818
build_py39(){

ci/wheel/excludes.params

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
--exclude libc10.so
2+
--exclude libc10_hip.so
3+
--exclude libtorch.so
4+
--exclude libtorch_hip.so
5+
--exclude libtorch_cpu.so
6+
--exclude libtorch_python.so

0 commit comments

Comments
 (0)