Jiaming Yuan bc69a3e877

[EM] Improve memory estimation for quantile sketching. (#10843) ...

I- Add basic estimation for RMM.
- Re-estimate after every sub-batch.
- Some debug logs for memory usage.
- Fix the locking mechanism in the memory allocator logger.

2024-09-25 03:20:09 +08:00

..

c_api

[EM] Make page concatenation optional. (#10826)

2024-09-24 06:19:28 +08:00

collective

[EM] Refactor ellpack construction. (#10810)

2024-09-09 14:10:10 +08:00

common

[EM] Improve memory estimation for quantile sketching. (#10843)

2024-09-25 03:20:09 +08:00

data

[EM] Make page concatenation optional. (#10826)

2024-09-24 06:19:28 +08:00

gbm

Check correct dump format for gblinear. (#10831)

2024-09-21 00:32:52 +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] Make page concatenation optional. (#10826)

2024-09-24 06:19:28 +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