Hendrik Groove
58a27ba968
more logging
2024-10-20 20:59:23 +02:00
Hendrik Groove
c964dd62b4
more logging
2024-10-20 20:53:50 +02:00
Hendrik Groove
4a10135006
validate label debug
2024-10-20 18:11:03 +02:00
Hendrik Groove
f54355f470
fix path
2024-10-20 17:56:27 +02:00
Hendrik Groove
08f3936bc9
fix path
2024-10-20 17:51:05 +02:00
Hendrik Groove
f50d5344f3
get gradient error logging
2024-10-20 17:40:52 +02:00
Hendrik Groove
ab41cd26a6
add gpu error check
2024-10-20 17:34:51 +02:00
Hendrik Groove
fd95be5f20
validate label logging
2024-10-20 17:32:22 +02:00
Hendrik Groove
60a3bea7c6
add logging
2024-10-20 17:30:17 +02:00
Hendrik Groove
7301022fed
logging
2024-10-20 17:05:34 +02:00
Hendrik Groove
288193cf82
try
2024-10-20 02:41:57 +02:00
Hendrik Groove
e142b52540
use new func
2024-10-20 02:18:41 +02:00
Hendrik Groove
e8fceb8198
add logging
2024-10-20 02:03:55 +02:00
Hendrik Groove
971d3ca8cd
array interface
2024-10-20 01:46:48 +02:00
Hendrik Groove
206f305b65
array interface
2024-10-20 01:28:40 +02:00
Hendrik Groove
8e703f3a5a
try hipHostMalloc
2024-10-20 01:13:44 +02:00
Hendrik Groove
0790bf7f8f
change back
2024-10-17 17:47:10 +02:00
Hendrik Groove
d8a92fe783
test
2024-10-17 17:42:37 +02:00
Hui Liu
b27f35e270
rm hip from src
2024-04-22 12:31:14 -07:00
Hui Liu
8b75204fed
merge latest change from upstream
2024-04-22 09:35:31 -07:00
Jiaming Yuan
3fbb221fec
[coll] Implement shutdown for tracker and comm. ( #10208 )
...
- Force shutdown the tracker.
- Implement shutdown notice for error handling thread in comm.
2024-04-20 04:08:17 +08:00
Jiaming Yuan
3f64b4fde3
[coll] Add global functions. ( #10203 )
2024-04-19 03:17:23 +08:00
Jiaming Yuan
4b10200456
[coll] Improve event loop. ( #10199 )
...
- Add a test for blocking calls.
- Do not require the queue to be empty after waking up; this frees up the thread to answer blocking calls.
- Handle EOF in read.
- Improve the error message in the result. Allow concatenation of multiple results.
2024-04-18 03:29:52 +08:00
Jiaming Yuan
1022909bbe
Fix global config for external memory. ( #10173 )
...
Pass the thread-local configuration between threads.
2024-04-11 01:29:28 +08:00
Jiaming Yuan
f0a138f33a
Fix pyspark with verbosity=3. ( #10172 )
2024-04-09 23:18:56 +08:00
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
230010d9a0
Cleanup set info. ( #10139 )
...
- Use the array interface internally.
- Deprecate `XGDMatrixSetDenseInfo`.
- Deprecate `XGDMatrixSetUIntInfo`.
- Move the handling of `DataType` into the deprecated C function.
---------
Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
2024-03-26 23:26:24 +08:00
Hui Liu
ff549ae933
sync upstream code
2024-03-20 16:14:38 -07:00
Jiaming Yuan
ca4801f81d
Work with IPv6 in the new tracker. ( #10125 )
2024-03-20 05:19:23 +08:00
Jiaming Yuan
53fc17578f
Use std::uint64_t for row index. ( #10120 )
...
- Use std::uint64_t instead of size_t to avoid implementation-defined type.
- Rename to bst_idx_t, to account for other types of indexing.
- Small cleanup to the base header.
2024-03-15 18:43:49 +08:00
Jiaming Yuan
56b1868278
Fix compilation with the latest ctk. ( #10123 )
2024-03-15 08:04:41 +08:00
Hui Liu
968dbf25fb
merge latest changes
2024-03-12 09:13:09 -07:00
Jiaming Yuan
1450aebb74
Fix pairwise objective with NDCG metric along with custom gain. ( #10100 )
...
* Fix pairwise objective with NDCG metric.
- Allow setting `ndcg_exp_gain` for `rank:pairwise`.
This is useful when using pairwise for objective but ndcg for metric.
2024-03-11 14:54:10 +08:00
Jiaming Yuan
2c13f90384
Support graphviz plot for multi-target tree. ( #10093 )
2024-03-09 05:35:25 +08:00
Jiaming Yuan
e14c3b9325
Optional normalization for learning to rank. ( #10094 )
2024-03-08 12:41:21 +08:00
Jiaming Yuan
3941b31ade
Disable column sample by node for the exact tree method. ( #10083 )
2024-03-01 14:16:10 +08:00
Jiaming Yuan
8189126d51
Add CUDA iterator to tensor view. ( #10074 )
2024-03-01 14:15:31 +08:00
Jiaming Yuan
5ac233280e
Require context in aggregators. ( #10075 )
2024-02-28 03:12:42 +08:00
Jiaming Yuan
0ce4372bd4
Use UBJSON for serializing splits for vertical data split. ( #10059 )
2024-02-25 00:18:23 +08:00
Jiaming Yuan
2e4ea5ecc0
Support f64 for ubjson. ( #10055 )
2024-02-21 02:18:42 +08:00
Jiaming Yuan
d37b83e8d9
Fix UBJSON with boolean value. ( #10054 )
2024-02-20 22:13:51 +08:00
david-cortes
6e3c899ba7
[R] Don't cap global number of threads for serialization ( #10028 )
2024-02-20 11:13:00 +08:00
Louis Desreumaux
edf501d227
Implement contribution prediction with QuantileDMatrix ( #10043 )
...
---------
Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
2024-02-19 21:03:29 +08:00
david-cortes
a730c7e67e
[R] allow using seed with regular RNG ( #10029 )
2024-02-04 16:22:22 +08:00
Hui Liu
44db1cef54
Merge branch 'master' into sync-2024Jan24
2024-02-01 14:41:48 -08:00
Philip Hyunsu Cho
4dfbe2a893
[CI] Test building for 32-bit arch ( #10021 )
...
* [CI] Test building for 32-bit arch
* Update CMakeLists.txt
* Fix yaml
* Use Debian container
* Remove -Werror for 32-bit
* Revert "Remove -Werror for 32-bit"
This reverts commit c652bc6a037361bcceaf56fb01863210b462793d.
* Don't error for overloaded-virtual warning
* Ignore some warnings from dmlc-core
* Fix compiler warnings
* Fix formatting
* Apply suggestions from code review
Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
* Add more cast
---------
Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
2024-01-31 13:20:51 -08:00
Hui Liu
2cb579ff3c
fix memory type
2024-01-26 15:46:42 -08:00
Jiaming Yuan
a76d6c6131
Fix cpp deprecation. ( #10010 )
2024-01-26 02:13:40 +08:00
Philip Hyunsu Cho
c8f5d190c6
[CI] Stop Windows pipeline upon a failing pytest ( #10003 )
2024-01-24 22:54:21 -08:00
Hui Liu
3fe874078c
merge latest changes
2024-01-24 13:30:08 -08:00