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

- ranking metric acceleration on the gpu #5398

Merged
merged 21 commits into from
Mar 22, 2020
Merged

- ranking metric acceleration on the gpu #5398

Show file tree
Hide file tree
Changes from 11 commits
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
604c63b
- ranking metric accelaration on the gpu
sriramch Mar 6, 2020
450c2f7
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 7, 2020
810e537
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 7, 2020
06f268e
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 10, 2020
987b241
- accelerate the auc metric on gpu
sriramch Mar 10, 2020
6a85632
Revert "Sketching from adapters (#5365)"
sriramch Mar 10, 2020
2bc60f9
Revert "Revert "Sketching from adapters (#5365)""
sriramch Mar 11, 2020
e4a4803
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 11, 2020
3c319ed
- fix bugs and add tests to detect them
sriramch Mar 11, 2020
dac5a95
- use sensible labels in test
sriramch Mar 11, 2020
5e70e6c
- add more exception scenario tests
sriramch Mar 11, 2020
82b58ad
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 12, 2020
43f411d
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 14, 2020
0f04285
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 15, 2020
d754882
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 15, 2020
5ee54ba
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 17, 2020
e74fd13
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 19, 2020
509d299
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 19, 2020
3290cc8
Merge branch 'master' of https://github.com/dmlc/xgboost into gpu_ran…
sriramch Mar 20, 2020
8416aab
- release intermediary vectors to free up space for more ones to reli…
sriramch Mar 20, 2020
d502df4
- ignore lint error
sriramch Mar 20, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions src/metric/metric.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -81,5 +81,8 @@ namespace metric {
DMLC_REGISTRY_LINK_TAG(elementwise_metric);
DMLC_REGISTRY_LINK_TAG(multiclass_metric);
DMLC_REGISTRY_LINK_TAG(rank_metric);
#ifdef XGBOOST_USE_CUDA
DMLC_REGISTRY_LINK_TAG(rank_metric_gpu);
#endif
} // namespace metric
} // namespace xgboost
2 changes: 1 addition & 1 deletion src/metric/rank_metric.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -260,7 +260,7 @@ struct EvalAuc : public Metric {
!info.group_ptr_.empty() && info.weights_.Size() != info.num_row_;

// Check if we have a GPU assignment; else, revert back to CPU
if (tparam_->gpu_id >= 0 && is_ranking_task) {
if (tparam_->gpu_id >= 0) {
if (!auc_gpu_) {
// Check and see if we have the GPU metric registered in the internal registry
auc_gpu_.reset(GPUMetric::CreateGPUMetric(this->Name(), tparam_));
Expand Down
Loading