-
Notifications
You must be signed in to change notification settings - Fork 540
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
[REVIEW] NVTX Markers for RF and RF-backend #3014
[REVIEW] NVTX Markers for RF and RF-backend #3014
Conversation
venkywonka
commented
Oct 19, 2020
- This PR adds NVTX Markers to major time-consuming function calls of the regressors and classifiers of RF and DecisionTrees.
- They span both RandomForest and DecisionTree code-bases
* this is to troubleshoot the bug in the `generateNextColor` function
…nh-fea-rf-nvtx-markers
Can one of the admins verify this patch? |
Please update the changelog in order to start CI tests. View the gpuCI docs here. |
…nh-ext-rf-nvtx-markers
Codecov Report
@@ Coverage Diff @@
## branch-0.18 #3014 +/- ##
===============================================
+ Coverage 71.48% 71.75% +0.26%
===============================================
Files 207 212 +5
Lines 16748 17089 +341
===============================================
+ Hits 11973 12262 +289
- Misses 4775 4827 +52
Continue to review full report at Codecov.
|
@venkywonka can you take a look at the merge conflicts? |
… enh-ext-rf-nvtx-markers
done @JohnZed ✌🏾 |
Will update copyrights shortly |
@venkywonka, can you try updating your branch? The PR below just merged which should get |
… enh-ext-rf-nvtx-markers
…onka/cuml into enh-ext-rf-nvtx-markers
CHANGELOG.md
Outdated
@@ -136,6 +136,7 @@ Please see https://github.com/rapidsai/cuml/releases/tag/branch-0.18-latest for | |||
- PR #2916: Add SKLearn multi-class GBDT model support in FIL | |||
|
|||
## Improvements | |||
- PR #3014: NVTX Markers for RF and RF-backend |
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.
please make sure to remove this entry since 0.16
has already been released. Manual changelog entries are no longer required, so you can just omit any changes to this file entirely.
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.
my bad @ajschmidt8 !, taken it out ✌🏾
@@ -418,6 +419,8 @@ class RandomForestRegressor(BaseRandomForestModel, RegressorMixin): | |||
Perform Random Forest Regression on the input data | |||
|
|||
""" | |||
nvtx_range_push("Fit RF-Regressor @randomforestregressor.pyx") |
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 really add a context manager so we can do: ```
with nvtx_range("fit"):
.. code here
and not need to remember to pop (or have to manage pop correctly in the case of exceptions or early returns)
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.
agreed @JohnZed ✌🏾
@gpucibot merge |