-
Notifications
You must be signed in to change notification settings - Fork 550
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
Configurable libcuml++ per algorithm #4296
Configurable libcuml++ per algorithm #4296
Conversation
This PR has been labeled |
rerun tests |
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.
Looks great! I requested one additional comment/docstring tweak, but it's fine to merge without.
cpp/CMakeLists.txt
Outdated
message(VERBOSE "CUML: Build and statically link FAISS library: ${CUML_USE_FAISS_STATIC}") | ||
message(VERBOSE "CUML: Build and statically link Treelite library: ${CUML_USE_TREELITE_STATIC}") | ||
|
||
set(CUML_ALGORITHMS "ALL" CACHE STRING "Experimental: Choose which algorithms are built into libcuml++.so.") |
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.
Can we note here (even just in a comment) that this is a semicolon-separated list of all-caps strings?
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.
To be fair, they don't need to be all caps, they can be any capitalization you want
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.
We should probably update the documentation as well to reflect the new building options (and eventually enable it in the build.sh
), but otherwise it LGTM.
Codecov Report
@@ Coverage Diff @@
## branch-22.04 #4296 +/- ##
===============================================
Coverage ? 83.88%
===============================================
Files ? 250
Lines ? 20062
Branches ? 0
===============================================
Hits ? 16829
Misses ? 3233
Partials ? 0
Flags with carried forward coverage won't be shown. Click here to find out more. Continue to review full report at Codecov.
|
@gpucibot merge |
Depends on rapidsai#4295 PR allows `libcuml++` to be built with individual algorithms, or individual families of algorithms with the argument `CUML_ALGORITHMS`. It defaults to `ALL`, and can take multiple options like: ``` cmake .. -DCUML_ALGORITHMS="FIL;TREESHAP" ``` which will build a `libcuml++` only containing FIL and GPUTreeSHAP components. PR to update build documentation will follow up. Authors: - Dante Gama Dessavre (https://github.com/dantegd) - Divye Gala (https://github.com/divyegala) Approvers: - William Hicks (https://github.com/wphicks) - Corey J. Nolet (https://github.com/cjnolet) URL: rapidsai#4296
Depends on #4295
PR allows
libcuml++
to be built with individual algorithms, or individual families of algorithms with the argumentCUML_ALGORITHMS
. It defaults toALL
, and can take multiple options like:which will build a
libcuml++
only containing FIL and GPUTreeSHAP components.PR to update build documentation will follow up.