Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
xgboost/src/collective
History
Jiaming Yuan 8fe1a2213c
Cleanup code for distributed training. (#9805) 
* Cleanup code for distributed training.

- Merge `GetNcclResult` into nccl stub.
- Split up utilities from the main dask module.
- Let Channel return `Result` to accommodate nccl channel.
- Remove old `use_label_encoder` parameter.
2023-11-25 09:10:56 +08:00
..
aggregator.cuh
Remove internal use of gpu_id. (#9568)
2023-09-20 23:29:51 +08:00
aggregator.h
Support vertical federated learning with gpu_hist (#9539)
2023-09-03 11:37:11 +08:00
allgather.cc
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
allgather.h
[coll] Add nccl. (#9726)
2023-10-28 16:33:58 +08:00
allreduce.cc
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
allreduce.h
Serialize expand entry for allgather. (#9702)
2023-10-24 14:33:28 +08:00
broadcast.cc
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
broadcast.h
[coll] Broadcast. (#9659)
2023-10-14 09:34:37 +08:00
coll.cc
[coll] Improvements and fixes for tracker and allreduce. (#9745)
2023-11-02 04:06:46 +08:00
coll.cu
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
coll.cuh
Use dlopen to load NCCL. (#9796)
2023-11-22 19:27:31 +08:00
coll.h
[coll] Add federated coll. (#9738)
2023-11-01 04:06:46 +08:00
comm_group.cc
Use dlopen to load NCCL. (#9796)
2023-11-22 19:27:31 +08:00
comm_group.h
Use dlopen to load NCCL. (#9796)
2023-11-22 19:27:31 +08:00
comm.cc
Use dlopen to load NCCL. (#9796)
2023-11-22 19:27:31 +08:00
comm.cu
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
comm.cuh
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
comm.h
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
communicator-inl.cuh
Support column split in GPU evaluate splits (#9511)
2023-08-23 16:33:43 +08:00
communicator-inl.h
Serialize expand entry for allgather. (#9702)
2023-10-24 14:33:28 +08:00
communicator.cc
Use dlopen to load NCCL. (#9796)
2023-11-22 19:27:31 +08:00
communicator.cu
Use dlopen to load NCCL. (#9796)
2023-11-22 19:27:31 +08:00
communicator.h
Use dlopen to load NCCL. (#9796)
2023-11-22 19:27:31 +08:00
device_communicator_adapter.cuh
Improve allgather functions (#9649)
2023-10-12 23:31:43 +08:00
device_communicator.cuh
Support column split in GPU evaluate splits (#9511)
2023-08-23 16:33:43 +08:00
in_memory_communicator.cc
Add an in-memory collective communicator (#8494)
2022-12-01 00:24:12 +08:00
in_memory_communicator.h
Improve allgather functions (#9649)
2023-10-12 23:31:43 +08:00
in_memory_handler.cc
Improve allgather functions (#9649)
2023-10-12 23:31:43 +08:00
in_memory_handler.h
Improve allgather functions (#9649)
2023-10-12 23:31:43 +08:00
loop.cc
Implement column sampler in CUDA. (#9785)
2023-11-17 04:29:08 +08:00
loop.h
Implement column sampler in CUDA. (#9785)
2023-11-17 04:29:08 +08:00
nccl_device_communicator.cu
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
nccl_device_communicator.cuh
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
nccl_stub.cc
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
nccl_stub.h
Cleanup code for distributed training. (#9805)
2023-11-25 09:10:56 +08:00
noop_communicator.h
Improve allgather functions (#9649)
2023-10-12 23:31:43 +08:00
protocol.h
[coll] Implement a new tracker and a communicator. (#9650)
2023-10-12 12:49:16 +08:00
rabit_communicator.h
Support column-wise data split with in-memory inputs (#9628)
2023-10-17 12:16:39 +08:00
socket.cc
Build a simple event loop for collective. (#9593)
2023-09-20 02:09:07 +08:00
tracker.cc
[coll] Add comm group. (#9759)
2023-11-07 11:12:31 +08:00
tracker.h
[coll] Add C API for the tracker. (#9773)
2023-11-08 18:17:14 +08:00