fix lib paths (still not to update rpath/runpath)

ci/build_wheel.sh

@@ -55,6 +55,7 @@ mkdir -p final_dist
 python -m auditwheel repair \
     -w final_dist \
     --exclude "libucm.so.0" \
+    --exclude "libucp.so.0" \
     --exclude "libucx.so.0" \
     --exclude "libucs.so.0" \
     --exclude "libuct.so.0" \

pyproject.toml

@@ -22,7 +22,7 @@ requires = [
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project]
-name = "ucx-py"
+name = "ucx-py-cu12"
 dynamic = ["version"]
 description = "Python Bindings for the Unified Communication X library (UCX)"
 readme = { file = "README.md", content-type = "text/markdown" }

setup.py

@@ -12,8 +12,24 @@
 from setuptools import setup
 from setuptools.extension import Extension
 
-include_dirs = [os.path.dirname(get_python_inc())]
 library_dirs = [get_config_var("LIBDIR")]
+
+# if the 'libucx' wheel was installed, add its library location to
+# library_dirs so it'll be linked against
+#
+# NOTE: doing this '.__file__' stuff because `sysconfig.get_path("platlib")` and similar
+#       can return paths to the main Python interpreters' installation locations...
+#       not where 'libucx' is going to be installed at build time when using
+#       build isolation
+try:
+    import libucx
+
+    libucx_libdir = os.path.join(os.path.dirname(libucx.__file__), "lib")
+    library_dirs.append(libucx_libdir)
+except ImportError:
+    pass
+
+include_dirs = [os.path.dirname(get_python_inc())]
 libraries = ["ucp", "uct", "ucm", "ucs"]
 extra_compile_args = ["-std=c99", "-Werror"]