Fix `rmm ._lib` imports

rapidsai / rmm

RAPIDS Memory Manager

https://docs.rapids.ai/api/rmm/stable/

Apache License 2.0

495 stars 198 forks source link

Fix `rmm ._lib` imports #1693

Closed Matt711 closed 1 month ago

Matt711 commented 1 month ago

Description

This PR fixes a bug in #1676. It makes sure that rmm imports work correctly using both from rmm._lib... and import rmm._lib... syntax.

Checklist

[ ] I am familiar with the Contributing Guidelines.
[ ] New or existing tests cover these changes.
[ ] The documentation is up to date with these changes.

vyasr commented 1 month ago

I'm going to go ahead and merge because this is currently breaking some users. If we have more changes that we want, we can address them in follow-ups, but I'd like to be able to experiment ASAP to make sure that the underlying issues are resolved in all cases.

vyasr commented 1 month ago

/merge