Jiaming Yuan 61dd854a52

[EM] Refactor GPU histogram builder. (#10764) ...

- Expose the maximum number of cached nodes to be consistent with the CPU implementation. Also easier for testing.
- Extract the subtraction trick for easier testing.
- Split up the `GradientQuantiser` to avoid circular dependency.

2024-08-30 02:39:14 +08:00

..

c_api

Use std::uint64_t for row index. (#10120)

2024-03-15 18:43:49 +08:00

collective

Increase timeout in C++ tests from 1 to 5 seconds. (#10756)

2024-08-28 02:27:14 +08:00

common

Increase timeout in C++ tests from 1 to 5 seconds. (#10756)

2024-08-28 02:27:14 +08:00

data

[EM] Return a full DMatrix instead of a Ellpack from the GPU sampler. (#10753)

2024-08-28 01:05:11 +08:00

gbm

[EM] Support SHAP contribution with QDM. (#10724)

2024-08-22 05:25:10 +08:00

linear

Use matrix for gradient. (#9508)

2023-08-24 05:29:52 +08:00

metric

Revamp the rabit implementation. (#10112)

2024-05-20 11:56:23 +08:00

objective

[EM] Support SHAP contribution with QDM. (#10724)

2024-08-22 05:25:10 +08:00

plugin

[SYCL] Implement UpdatePredictionCache and connect updater with leraner. (#10701)

2024-08-22 02:07:44 +08:00

predictor

[EM] Support SHAP contribution with QDM. (#10724)

2024-08-22 05:25:10 +08:00

tree

[EM] Refactor GPU histogram builder. (#10764)

2024-08-30 02:39:14 +08:00

categorical_helpers.h

Use quantised gradients in gpu_hist histograms (#8246)

2022-09-26 17:35:35 +02:00

CMakeLists.txt

[fed] Split up federated test CMake file. (#10566)

2024-07-11 13:09:18 +08:00

filesystem.h

[EM] Support mmap backed ellpack. (#10602)

2024-07-18 08:20:21 +08:00

helpers.cc

[EM] Support SHAP contribution with QDM. (#10724)

2024-08-22 05:25:10 +08:00

helpers.cu

[EM] Enable access to the number of batches. (#10691)

2024-08-17 02:59:45 +08:00

helpers.h

[EM] Support SHAP contribution with QDM. (#10724)

2024-08-22 05:25:10 +08:00

histogram_helpers.h

[EM] Support mmap backed ellpack. (#10602)

2024-07-18 08:20:21 +08:00

objective_helpers.cc

Revamp the rabit implementation. (#10112)

2024-05-20 11:56:23 +08:00

objective_helpers.h

Revamp the rabit implementation. (#10112)

2024-05-20 11:56:23 +08:00

test_cache.cc

Fix build and C++ tests for FreeBSD (#10480)

2024-06-28 01:47:55 -07:00

test_context.cc

Add 'sycl' devices to the context (#9691)

2023-10-26 22:17:56 +08:00

test_context.cu

[EM] Support mmap backed ellpack. (#10602)

2024-07-18 08:20:21 +08:00

test_global_config.cc

Add use_rmm flag to global configuration (#6656)

2021-03-09 14:53:05 -08:00

test_helpers.cc

[EM] Allow staging ellpack on host for GPU external memory. (#10488)

2024-06-28 04:42:18 +08:00

test_learner.cc

[coll] Improve column split tests with named threads. (#10735)

2024-08-24 12:43:47 +08:00

test_logging.cc

Remove silent from R demos. (#5675)

2020-05-19 18:20:46 +08:00

test_main.cc

Revamp the rabit implementation. (#10112)

2024-05-20 11:56:23 +08:00

test_multi_target.cc

Use matrix for gradient. (#9508)

2023-08-24 05:29:52 +08:00

test_serialization.cc

Support f64 for ubjson. (#10055)

2024-02-21 02:18:42 +08:00