Directly import and load librmm

python/libcuvs/libcuvs/load.py

@@ -46,19 +46,18 @@ def _load_wheel_installation(soname: str):
 def load_library():
     """Dynamically load libcuvs.so and its dependencies"""
     try:
-        # libraft must be loaded before libcuvs because libcuvs
-        # references its symbols
+        # These libraries  must be loaded before libcuvs because libcuvs
+        # references their symbols
         import libraft
+        import librmm
 
+        librmm.load_library()
         libraft.load_library()
     except ModuleNotFoundError:
-        # 'libcuvs' has a runtime dependency on 'libraft'. However,
-        # that dependency might be satisfied by the 'libraft' conda package
-        # (which does not have any Python modules), instead of the
-        # 'libraft' wheel.
-        #
-        # In that situation, assume that 'libraft.so' is in a place where
-        # the loader can find it.
+        # These runtime dependencies might be satisfied by conda packages
+        # (which do not have any Python modules) instead of wheels. In that
+        # situation, assume that the libraries are in a place where the loader
+        # can find them.
         pass
 
     prefer_system_installation = (