xgboost

Hendrik/xgboost

Fork 0

Commit Graph

Author	SHA1	Message	Date
Jiaming Yuan	8bad677c2f	Update collective implementation. (#10152 ) * Update collective implementation. - Cleanup resource during `Finalize` to avoid handling threads in destructor. - Calculate the size for allgather automatically. - Use simple allgather for small (smaller than the number of worker) allreduce.	2024-03-30 18:57:31 +08:00
Jiaming Yuan	bc995a4865	[coll] Add federated coll. (#9738 ) - Define a new data type, the proto file is copied for now. - Merge client and communicator into `FederatedColl`. - Define CUDA variant. - Migrate tests for CPU, add tests for CUDA.	2023-11-01 04:06:46 +08:00

Author

SHA1

Message

Date

Jiaming Yuan

8bad677c2f

Update collective implementation. (#10152 )

* Update collective implementation.

- Cleanup resource during `Finalize` to avoid handling threads in destructor.
- Calculate the size for allgather automatically.
- Use simple allgather for small (smaller than the number of worker) allreduce.

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

Jiaming Yuan

bc995a4865

[coll] Add federated coll. (#9738 )

- Define a new data type, the proto file is copied for now.
- Merge client and communicator into `FederatedColl`.
- Define CUDA variant.
- Migrate tests for CPU, add tests for CUDA.

2023-11-01 04:06:46 +08:00

2 Commits