Fix `rmm ._lib` imports

rapidsai / rmm

RAPIDS Memory Manager

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

Apache License 2.0

478 stars 195 forks source link

Fix `rmm ._lib` imports #1693

Closed Matt711 closed 14 hours ago

Matt711 commented 1 day 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.

I'm adding DO NOT MERGE until I do some more testing.

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 14 hours 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 14 hours ago

/merge