-
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
[ENH] [5/5] Header structure: isolate logger and memory pool #1441
[ENH] [5/5] Header structure: isolate logger and memory pool #1441
Conversation
Instead of having the specializations in sub-directories, the raft_runtime source files now mimic the include/ directory hierarchy.
Used .data() instead of .data_handle()
These types are not used in the ext header, but are useful to have.
Under multiple combinations of RAFT_EXPLICIT_INSTANTIATE_ONLY and RAFT_COMPILED
The compute_similarity and interleaved_scan kernel are quite expensive to compile. Splitting the headers in this commit.
These two include files are likely to transitively include spdlog, which increases compilation times.
Note the difference in CI time for this PR compared to the first on in the series:
|
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.
Reviewed the last commit, it looks good to me.
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, pending changes in #1437. I verified the logger changes.
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.
Sorry, @ahendriksen, I looked a little more closely at the macros and I'd like to consolidate these so we don't end up with too much duplication for these core bits. Otherwise, I think this looks great.
This is a rebase of all the commits in PRs: - #1437 - #1438 - #1439 - #1440 - #1441 The original PRs have not been rebased to preserve review comments. This PR is up to date with branch 23.06. Closes #1416 Authors: - Allard Hendriksen (https://github.com/ahendriksen) Approvers: - Divye Gala (https://github.com/divyegala) - Corey J. Nolet (https://github.com/cjnolet) URL: #1469
This PR isolates the logger and memory pool. These are likely to transitively include spdlog, which slows down compile times. After rapidsai/rmm#1241 is merged, this will yield a substantial improvement in compile times for smaller translation units.
It is a follow up to PR #1440.