Rong Ou a2686543a9

Common interface for collective communication (#8057) ...

* implement broadcast for federated communicator

* implement allreduce

* add communicator factory

* add device adapter

* add device communicator to factory

* add rabit communicator

* add rabit communicator to the factory

* add nccl device communicator

* add synchronize to device communicator

* add back print and getprocessorname

* add python wrapper and c api

* clean up types

* fix non-gpu build

* try to fix ci

* fix std::size_t

* portable string compare ignore case

* c style size_t

* fix lint errors

* cross platform setenv

* fix memory leak

* fix lint errors

* address review feedback

* add python test for rabit communicator

* fix failing gtest

* use json to configure communicators

* fix lint error

* get rid of factories

* fix cpu build

* fix include

* fix python import

* don't export collective.py yet

* skip collective communicator pytest on windows

* add review feedback

* update documentation

* remove mpi communicator type

* fix tests

* shutdown the communicator separately

Co-authored-by: Hyunsu Cho <chohyu01@cs.washington.edu>

2022-09-12 15:21:12 -07:00

..

algorithm.cuh

Copy data from Ellpack to GHist. (#8215)

2022-09-06 23:05:49 +08:00

algorithm.h

Copy data from Ellpack to GHist. (#8215)

2022-09-06 23:05:49 +08:00

base64.h

Revert "Reorder includes. (#5749)" (#5771)

2020-06-09 10:29:28 -07:00

bitfield.h

Fix index type for bitfield. (#7541)

2022-01-05 19:23:29 +08:00

categorical.h

Implement max_cat_threshold for CPU. (#7957)

2022-06-04 11:02:46 +08:00

charconv.cc

Fix spelling in documents (#6948)

2021-05-11 20:44:36 +08:00

charconv.h

Implement fast number serialization routines. (#5772)

2020-06-17 12:39:23 +08:00

column_matrix.cc

Split up column matrix initialization. (#8060)

2022-07-14 10:34:47 +08:00

column_matrix.h

Copy data from Ellpack to GHist. (#8215)

2022-09-06 23:05:49 +08:00

common.cc

Make HostDeviceVector single gpu only (#4773)

2019-08-26 09:51:13 +12:00

common.cu

[Breaking] Accept multi-dim meta info. (#7405)

2021-11-18 23:02:54 +08:00

common.h

Enable distributed GPU training over Rabit (#7930)

2022-05-31 04:09:45 +08:00

compressed_iterator.h

Revert "Reorder includes. (#5749)" (#5771)

2020-06-09 10:29:28 -07:00

config.h

[breaking] Change internal model serialization to UBJSON. (#7556)

2022-01-16 02:11:53 +08:00

device_helpers.cu

Update CUDA docker image and NCCL. (#8139)

2022-08-07 16:32:41 +08:00

device_helpers.cuh

Copy data from Ellpack to GHist. (#8215)

2022-09-06 23:05:49 +08:00

group_data.h

Memory consumption fix for row-major adapters (#6779)

2021-03-26 08:44:30 +08:00

hist_util.cc

Optimization/buildhist/hist util (#8218)

2022-09-02 19:39:45 +08:00

hist_util.cu

Handle missing values in dataframe with category dtype. (#7331)

2021-10-28 03:33:54 +08:00

hist_util.cuh

Support CPU input for device QuantileDMatrix. (#8136)

2022-08-11 21:21:26 +08:00

hist_util.h

Copy data from Ellpack to GHist. (#8215)

2022-09-06 23:05:49 +08:00

host_device_vector.cc

Add minimal emscripten build support (#7954)

2022-05-30 14:11:40 +08:00

host_device_vector.cu

Implement typed storage for tensor. (#7429)

2021-11-14 18:53:13 +08:00

io.cc

[breaking] Change internal model serialization to UBJSON. (#7556)

2022-01-16 02:11:53 +08:00

io.h

Common interface for collective communication (#8057)

2022-09-12 15:21:12 -07:00

json.cc

Fix compiler warnings. (#8022)

2022-06-22 21:29:10 +08:00

linalg_op.cuh

[POC] Experimental support for l1 error. (#7812)

2022-04-26 21:41:55 +08:00

linalg_op.h

Implement slope for Pseduo-Huber. (#7727)

2022-03-14 21:42:38 +08:00

math.h

Implement slope for Pseduo-Huber. (#7727)

2022-03-14 21:42:38 +08:00

numeric.h

Extract partial sum into an independent function. (#7889)

2022-05-13 14:30:35 +08:00

observer.h

Unify evaluation functions. (#6037)

2020-08-26 14:23:27 +08:00

partition_builder.h

docs: fix simple typo, cannonical -> canonical (#8099)

2022-07-20 21:04:50 +08:00

probability_distribution.h

GPU implementation of AFT survival objective and metric (#5714)

2020-07-17 01:18:13 -07:00

pseudo_huber.cc

Implement slope for Pseduo-Huber. (#7727)

2022-03-14 21:42:38 +08:00

pseudo_huber.h

Implement slope for Pseduo-Huber. (#7727)

2022-03-14 21:42:38 +08:00

quantile.cc

Small cleanups to various data types. (#8086)

2022-07-18 22:39:36 +08:00

quantile.cu

Fix compiler warnings. (#8022)

2022-06-22 21:29:10 +08:00

quantile.cuh

Implement categorical data support for SHAP. (#7053)

2021-06-25 19:02:46 +08:00

quantile.h

Small cleanups to various data types. (#8086)

2022-07-18 22:39:36 +08:00

random.cc

Fix feature weights with multiple column sampling. (#8100)

2022-07-22 20:23:05 +08:00

random.h

Small cleanup to hist tree method. (#7735)

2022-03-20 03:44:55 +08:00

ranking_utils.cuh

Remove use of device_idx in dh::LaunchN. (#7063)

2021-06-29 11:37:26 +08:00

row_set.h

[POC] Experimental support for l1 error. (#7812)

2022-04-26 21:41:55 +08:00

stats.cuh

[POC] Experimental support for l1 error. (#7812)

2022-04-26 21:41:55 +08:00

stats.h

[POC] Experimental support for l1 error. (#7812)

2022-04-26 21:41:55 +08:00

survival_util.cc

GPU implementation of AFT survival objective and metric (#5714)

2020-07-17 01:18:13 -07:00

survival_util.h

Fix spelling in documents (#6948)

2021-05-11 20:44:36 +08:00

threading_utils.cc

Fix file descriptor leak. (#7704)

2022-02-25 17:49:33 +08:00

threading_utils.h

Small cleanup to hist tree method. (#7735)

2022-03-20 03:44:55 +08:00

timer.cc

Remove synchronization in monitor. (#7164)

2021-08-11 16:33:53 +08:00

timer.h

Remove synchronization in monitor. (#7164)

2021-08-11 16:33:53 +08:00

transform.h

Remove omp_get_max_threads in objective. (#7589)

2022-01-24 04:35:49 +08:00

version.cc

Fix spelling in documents (#6948)

2021-05-11 20:44:36 +08:00

version.h

Fix spelling in documents (#6948)

2021-05-11 20:44:36 +08:00