keep the so number for xpu dependency

pytorch · Sep 16, 2024 · c861a31 · c861a31
1 parent ab63f3e
commit c861a31
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/manywheel/build_common.sh b/manywheel/build_common.sh
@@ -353,8 +353,11 @@ for pkg in /$WHEELHOUSE_DIR/torch_no_python*.whl /$WHEELHOUSE_DIR/torch*linux*.w
             fi
 
             # ROCm workaround for roctracer dlopens
-            if [[ "$DESIRED_CUDA" == *"rocm"* || "$DESIRED_CUDA" == *"xpu"* ]]; then
+            if [[ "$DESIRED_CUDA" == *"rocm"* ]]; then
                 patchedpath=$(fname_without_so_number $destpath)
+            # Keep the so number for XPU dependencies
+            elif [[ "$DESIRED_CUDA" == *"xpu"* ]]; then
+                patchedpath=$destpath
             else
                 patchedpath=$(fname_with_sha256 $destpath)
             fi