Skip to content
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 raft::span in TreeExplainer #4714

Merged
merged 30 commits into from
May 23, 2022
Merged

Conversation

hcho3
Copy link
Contributor

@hcho3 hcho3 commented Apr 27, 2022

Now that RAFT has a span implementation, use that in TreeExplainer.

@hcho3 hcho3 requested review from a team as code owners April 27, 2022 01:21
@github-actions github-actions bot added CUDA/C++ Cython / Python Cython or Python issue labels Apr 27, 2022
@hcho3 hcho3 added non-breaking Non-breaking change RAFT Issue or PR is related to RAFT improvement Improvement / enhancement to an existing function labels Apr 27, 2022
@github-actions github-actions bot removed the Cython / Python Cython or Python issue label May 11, 2022
@codecov-commenter
Copy link

Codecov Report

❗ No coverage uploaded for pull request base (branch-22.06@bc70681). Click here to learn what that means.
The diff coverage is n/a.

@@               Coverage Diff               @@
##             branch-22.06    #4714   +/-   ##
===============================================
  Coverage                ?   79.89%           
===============================================
  Files                   ?      180           
  Lines                   ?    11371           
  Branches                ?        0           
===============================================
  Hits                    ?     9085           
  Misses                  ?     2286           
  Partials                ?        0           
Flag Coverage Δ
dask 45.45% <0.00%> (?)
non-dask 69.51% <0.00%> (?)

Flags with carried forward coverage won't be shown. Click here to find out more.


Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update bc70681...ab7ef1f. Read the comment docs.

@hcho3
Copy link
Contributor Author

hcho3 commented May 11, 2022

@RAMitchell @trivialfis Can I get a review?

@@ -200,14 +171,30 @@ struct SplitCondition {
// Combine two split conditions on the same feature
__host__ __device__ void Merge(const SplitCondition& other)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Does merge ever get called on the host? Just wondering if we can remove the host categories.

Copy link
Member

@cjnolet cjnolet left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@cjnolet
Copy link
Member

cjnolet commented May 23, 2022

@gpucibot merge

@rapids-bot rapids-bot bot merged commit bc43c0e into rapidsai:branch-22.06 May 23, 2022
@hcho3 hcho3 deleted the remove_span branch May 23, 2022 16:49
vimarsh6739 pushed a commit to vimarsh6739/cuml that referenced this pull request Oct 9, 2023
Now that RAFT has a `span` implementation, use that in TreeExplainer.

Authors:
  - Philip Hyunsu Cho (https://github.com/hcho3)
  - Rory Mitchell (https://github.com/RAMitchell)

Approvers:
  - Jiaming Yuan (https://github.com/trivialfis)
  - Corey J. Nolet (https://github.com/cjnolet)

URL: rapidsai#4714
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CUDA/C++ improvement Improvement / enhancement to an existing function non-breaking Non-breaking change RAFT Issue or PR is related to RAFT
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

5 participants