Cleanup GPU Hist tests. (#10677)

* Cleanup GPU Hist tests.

- Remove GPU Hist gradient sampling test. The same properties are tested in the gradient
  sampler test suite.
- Move basic histogram tests into the histogram test suite.
- Remove the header inclusion of the `updater_gpu_hist.cu` in tests.
This commit is contained in:
Jiaming Yuan
2024-08-06 11:50:44 +08:00
committed by GitHub
parent 6ccf116601
commit cc3b56fc37
5 changed files with 104 additions and 183 deletions

View File

@@ -45,9 +45,7 @@
#include "xgboost/tree_model.h"
namespace xgboost::tree {
#if !defined(GTEST_TEST)
DMLC_REGISTRY_FILE_TAG(updater_gpu_hist);
#endif // !defined(GTEST_TEST)
// Manage memory for a single GPU
struct GPUHistMakerDevice {
@@ -831,13 +829,11 @@ class GPUHistMaker : public TreeUpdater {
std::shared_ptr<common::ColumnSampler> column_sampler_;
};
#if !defined(GTEST_TEST)
XGBOOST_REGISTER_TREE_UPDATER(GPUHistMaker, "grow_gpu_hist")
.describe("Grow tree with GPU.")
.set_body([](Context const* ctx, ObjInfo const* task) {
return new GPUHistMaker(ctx, task);
});
#endif // !defined(GTEST_TEST)
class GPUGlobalApproxMaker : public TreeUpdater {
public:
@@ -960,11 +956,9 @@ class GPUGlobalApproxMaker : public TreeUpdater {
common::Monitor monitor_;
};
#if !defined(GTEST_TEST)
XGBOOST_REGISTER_TREE_UPDATER(GPUApproxMaker, "grow_gpu_approx")
.describe("Grow tree with GPU.")
.set_body([](Context const* ctx, ObjInfo const* task) {
return new GPUGlobalApproxMaker(ctx, task);
});
#endif // !defined(GTEST_TEST)
} // namespace xgboost::tree