-
Notifications
You must be signed in to change notification settings - Fork 197
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Use FAISS with RMM #363
Use FAISS with RMM #363
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The changes are looking much better. I also want to make sure we clearly describe the path to removing the copied code and adapting it to their API once that issue is resolved. Once that's done, this PR will be ready to go.
Opened an issue on FAISS to avoid code duplication : facebookresearch/faiss#2097. Using a custom memory manager in FAISS currently requires overriding methods of It would be possible to prevent consequent rewrites thanks to updates to the The necessary changes would be the following :
|
@viclafargue, I looked back over my last comment and realized it was fairly ambiguous. I think having the comment on this PR is very useful as well but we should also have the comment in the code, preferably at the top of |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
This PR has been labeled |
@viclafargue it's been a little while so just wanted to check the status on this PR. IIRC, we are waiting on cuml side to pass successfuly, right? |
Thanks for reminding me of this. Yes, I just solved the merge conflicts on this PR. |
@gpucibot merge |
Answers rapidsai/cuml#2821