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

Optimize dart inplace predict perf. #6804

Merged
merged 2 commits into from
Mar 31, 2021

Conversation

trivialfis
Copy link
Member

Similar to optimization done in #6777 , but for in-place prediction.

@trivialfis
Copy link
Member Author

For 100 trees higgs:

  • Before
    GPU Inplace Predict::Duration 7.213954210281372

  • After
    GPU Inplace Predict::Duration 2.420344591140747

@trivialfis trivialfis merged commit 79b8b56 into dmlc:master Mar 31, 2021
@trivialfis trivialfis deleted the optimize-dart-inplace branch March 31, 2021 07:20
@trivialfis trivialfis mentioned this pull request Mar 31, 2021
8 tasks
trivialfis added a commit to trivialfis/xgboost that referenced this pull request Apr 6, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants