From ad7e3c93b19d2c4ce0b1223976917fb3ce0c038e Mon Sep 17 00:00:00 2001 From: Jeff Daily Date: Wed, 1 Oct 2025 00:02:34 +0000 Subject: [PATCH] [ROCm][CD] librocroller.so missing from ROCm 7 wheel (#164244) Pull Request resolved: https://github.com/pytorch/pytorch/pull/164244 Approved by: https://github.com/jeffdaily, https://github.com/Skylion007 Co-authored-by: Jeff Daily --- .ci/manywheel/build_rocm.sh | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/.ci/manywheel/build_rocm.sh b/.ci/manywheel/build_rocm.sh index ffc15bcdc5fa..bac56746f450 100755 --- a/.ci/manywheel/build_rocm.sh +++ b/.ci/manywheel/build_rocm.sh @@ -107,6 +107,10 @@ if [[ $ROCM_INT -ge 60200 ]]; then ROCM_SO_FILES+=("librocm-core.so") fi +if [[ $ROCM_INT -ge 70000 ]]; then + ROCM_SO_FILES+=("librocroller.so") +fi + OS_NAME=`awk -F= '/^NAME/{print $2}' /etc/os-release` if [[ "$OS_NAME" == *"CentOS Linux"* || "$OS_NAME" == *"AlmaLinux"* ]]; then LIBGOMP_PATH="/usr/lib64/libgomp.so.1"