xgboost

aggregator.cuh

Remove internal use of gpu_id. (#9568 )

2023-09-20 23:29:51 +08:00

aggregator.h

[coll] Add global functions. (#10203 )

2024-04-19 03:17:23 +08:00

allgather.cc

[coll] Add global functions. (#10203 )

2024-04-19 03:17:23 +08:00

allgather.h

[coll] Add global functions. (#10203 )

2024-04-19 03:17:23 +08:00

allreduce.cc

[coll] Add global functions. (#10203 )

2024-04-19 03:17:23 +08:00

allreduce.h

[coll] Add global functions. (#10203 )

2024-04-19 03:17:23 +08:00

broadcast.cc

Cleanup code for distributed training. (#9805 )

2023-11-25 09:10:56 +08:00

broadcast.h

[coll] Add global functions. (#10203 )

2024-04-19 03:17:23 +08:00

coll.cc

merge latest change from upstream

2024-04-22 09:35:31 -07:00

coll.cu

merge latest change from upstream

2024-04-22 09:35:31 -07:00

coll.cuh

Update collective implementation. (#10152 )

2024-03-30 18:57:31 +08:00

coll.h

Update collective implementation. (#10152 )

2024-03-30 18:57:31 +08:00

coll.hip

enable coll and comm

2023-10-30 15:15:05 -07:00

comm_group.cc

[coll] Implement shutdown for tracker and comm. (#10208 )

2024-04-20 04:08:17 +08:00

comm_group.h

[coll] Implement shutdown for tracker and comm. (#10208 )

2024-04-20 04:08:17 +08:00

comm.cc

merge latest change from upstream

2024-04-22 09:35:31 -07:00

comm.cu

merge latest change from upstream

2024-04-22 09:35:31 -07:00

comm.cuh

merge latest change from upstream

2024-04-22 09:35:31 -07:00

comm.h

[coll] Implement shutdown for tracker and comm. (#10208 )

2024-04-20 04:08:17 +08:00

comm.hip

enable coll and comm

2023-10-30 15:15:05 -07:00

communicator-inl.cc

Use UBJSON for serializing splits for vertical data split. (#10059 )

2024-02-25 00:18:23 +08:00

communicator-inl.cuh

Support column split in GPU evaluate splits (#9511 )

2023-08-23 16:33:43 +08:00

communicator-inl.h

Use UBJSON for serializing splits for vertical data split. (#10059 )

2024-02-25 00:18:23 +08:00

communicator.cc

merge latest changes

2023-12-13 21:06:28 -08:00

communicator.cu

merge latest changes

2023-12-13 21:06:28 -08:00

communicator.h

merge latest changes

2023-12-13 21:06:28 -08:00

communicator.hip

finished communicator.cu

2023-03-09 21:02:48 +01:00

device_communicator_adapter.cuh

sync upstream code

2024-03-20 16:14:38 -07: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

Update collective implementation. (#10152 )

2024-03-30 18:57:31 +08:00

loop.cc

[coll] Improve event loop. (#10199 )

2024-04-18 03:29:52 +08:00

loop.h

[coll] Improve event loop. (#10199 )

2024-04-18 03:29:52 +08:00

nccl_device_communicator.cu

merge latest change from upstream

2024-04-22 09:35:31 -07:00

nccl_device_communicator.cuh

merge latest changes

2023-12-13 21:06:28 -08:00

nccl_device_communicator.hip

add nccl_device_communicator.hip

2023-10-23 16:43:03 -07:00

nccl_stub.cc

fix random

2024-01-21 12:48:41 -08:00

nccl_stub.h

use __HIPCC__ for device code

2024-01-24 11:30:01 -08:00

noop_communicator.h

Improve allgather functions (#9649 )

2023-10-12 23:31:43 +08:00

protocol.h

[coll] Implement shutdown for tracker and comm. (#10208 )

2024-04-20 04:08:17 +08:00

rabit_communicator.h

Support column-wise data split with in-memory inputs (#9628 )

2023-10-17 12:16:39 +08:00

result.cc

[coll] Improve event loop. (#10199 )

2024-04-18 03:29:52 +08:00

socket.cc

[coll] Implement shutdown for tracker and comm. (#10208 )

2024-04-20 04:08:17 +08:00

tracker.cc

[coll] Implement shutdown for tracker and comm. (#10208 )

2024-04-20 04:08:17 +08:00

tracker.h

[coll] Implement shutdown for tracker and comm. (#10208 )

2024-04-20 04:08:17 +08:00