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.

Sign up for GitHub

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

Jump to bottom

Optimize GPU evaluation function for categorical data. #7705

Merged
merged 2 commits into from Feb 28, 2022
Merged

Optimize GPU evaluation function for categorical data. #7705

merged 2 commits into from Feb 28, 2022

Conversation

trivialfis
Copy link
Member

@trivialfis trivialfis commented Feb 25, 2022 •
edited

  • Use transform the pre-compute some of the values.

The time used for the cat-in-the-dat.py demo is reduced by about half from 4.2 seconds to about 2.1 seconds.

Suggested by @RAMitchell .

@trivialfis
Optimize GPU evaluation function for categorical data.
c0d7fb7 
* Use iterator to fetch the values for sort.
@trivialfis trivialfis mentioned this pull request Feb 25, 2022
Closed
@trivialfis trivialfis marked this pull request as draft February 26, 2022 01:37
@trivialfis
Copy link
Member Author

Unsure whether sorting a transform iterator is supported.

@trivialfis trivialfis marked this pull request as ready for review February 28, 2022 05:01
@trivialfis trivialfis merged commit 1d468e2 into dmlc:master Feb 28, 2022
@trivialfis trivialfis deleted the opt-gpu-eval branch February 28, 2022 09:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@RAMitchell RAMitchell RAMitchell approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
2 participants
@trivialfis @RAMitchell