Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
xgboost/src/common
History
Jiaming Yuan fedd9674c8
Implement column sampler in CUDA. (#9785) 
- CUDA implementation.
- Extract the broadcasting logic, we will need the context parameter after revamping the collective implementation.
- Some changes to the event loop for fixing a deadlock in CI.
- Move argsort into algorithms.cuh, add support for cuda stream.
2023-11-17 04:29:08 +08:00
..
algorithm.cuh
Implement column sampler in CUDA. (#9785)
2023-11-17 04:29:08 +08:00
algorithm.h
Specify char type in JSON. (#8949)
2023-03-22 19:13:44 +08:00
api_entry.h
Support exporting cut values (#9356)
2023-07-08 15:32:41 +08:00
base64.h
Revert "Reorder includes. (#5749)" (#5771)
2020-06-09 10:29:28 -07:00
bitfield.h
Serialize expand entry for allgather. (#9702)
2023-10-24 14:33:28 +08:00
categorical.h
Fix GPU categorical split memory allocation. (#9529)
2023-08-29 10:06:03 +08:00
charconv.cc
Restore clang tidy test. (#8861)
2023-03-03 13:47:04 -08:00
charconv.h
Restore clang tidy test. (#8861)
2023-03-03 13:47:04 -08:00
column_matrix.cc
Use ptr from mmap for GHistIndexMatrix and ColumnMatrix. (#9315)
2023-06-27 19:05:46 +08:00
column_matrix.h
Use weakref instead of id for DataIter cache. (#9445)
2023-08-10 00:40:06 +08:00
common.cc
Handle special characters in JSON model dump. (#9474)
2023-08-14 15:49:00 +08:00
common.cu
Remove MGPU cpp tests. (#8276)
2022-09-27 21:18:23 +08:00
common.h
[coll] Add federated coll. (#9738)
2023-11-01 04:06:46 +08:00
compressed_iterator.h
Restore clang tidy test. (#8861)
2023-03-03 13:47:04 -08:00
config.h
[breaking] Change internal model serialization to UBJSON. (#7556)
2022-01-16 02:11:53 +08:00
cuda_context.cuh
Skip optional synchronization in thrust. (#9212)
2023-05-30 17:23:09 +08:00
cuda_pinned_allocator.h
Work with newer thrust and libcudacxx (#8454)
2022-11-11 04:22:53 +08:00
deterministic.cuh
Extract floating point rounding routines. (#8771)
2023-02-12 04:26:41 +08:00
device_helpers.cuh
Implement column sampler in CUDA. (#9785)
2023-11-17 04:29:08 +08:00
error_msg.cc
Fix inplace predict with fallback when base margin is used. (#9536)
2023-09-05 01:04:24 +08:00
error_msg.h
[coll] Add C API for the tracker. (#9773)
2023-11-08 18:17:14 +08:00
group_data.h
Memory consumption fix for row-major adapters (#6779)
2021-03-26 08:44:30 +08:00
hist_util.cc
[coll] Pass context to various functions. (#9772)
2023-11-08 09:54:05 +08:00
hist_util.cu
[coll] Pass context to various functions. (#9772)
2023-11-08 09:54:05 +08:00
hist_util.cuh
Remove internal use of gpu_id. (#9568)
2023-09-20 23:29:51 +08:00
hist_util.h
Bound the size of the histogram cache. (#9440)
2023-08-08 03:21:26 +08:00
host_device_vector.cc
[coll] allgatherv. (#9688)
2023-10-19 03:13:50 +08:00
host_device_vector.cu
[coll] allgatherv. (#9688)
2023-10-19 03:13:50 +08:00
io.cc
Drop support for loading remote files. (#9504)
2023-08-21 23:34:05 +08:00
io.h
Extract JSON utils. (#9645)
2023-10-10 07:15:14 +08:00
json_utils.h
Extract JSON utils. (#9645)
2023-10-10 07:15:14 +08:00
json.cc
Handle special characters in JSON model dump. (#9474)
2023-08-14 15:49:00 +08:00
linalg_op.cuh
Reorder if-else statements to allow using of cpu branches for sycl-devices (#9682)
2023-10-18 10:55:33 +08:00
linalg_op.h
Reorder if-else statements to allow using of cpu branches for sycl-devices (#9682)
2023-10-18 10:55:33 +08:00
math.h
Fix clang warnings. (#9447)
2023-08-09 15:34:45 +08:00
numeric.cc
Reorder if-else statements to allow using of cpu branches for sycl-devices (#9682)
2023-10-18 10:55:33 +08:00
numeric.cu
Remove internal use of gpu_id. (#9568)
2023-09-20 23:29:51 +08:00
numeric.h
Handle special characters in JSON model dump. (#9474)
2023-08-14 15:49:00 +08:00
observer.h
Unify evaluation functions. (#6037)
2020-08-26 14:23:27 +08:00
optional_weight.h
Reorder if-else statements to allow using of cpu branches for sycl-devices (#9682)
2023-10-18 10:55:33 +08:00
partition_builder.h
Remove unused variables. (#9210)
2023-05-28 05:24:15 +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_loss_utils.cc
Extract JSON utils. (#9645)
2023-10-10 07:15:14 +08:00
quantile_loss_utils.h
Add quantile metric. (#8761)
2023-02-13 19:07:40 +08:00
quantile.cc
[coll] Pass context to various functions. (#9772)
2023-11-08 09:54:05 +08:00
quantile.cu
[coll] Pass context to various functions. (#9772)
2023-11-08 09:54:05 +08:00
quantile.cuh
[coll] Pass context to various functions. (#9772)
2023-11-08 09:54:05 +08:00
quantile.h
[coll] Pass context to various functions. (#9772)
2023-11-08 09:54:05 +08:00
random.cc
Implement column sampler in CUDA. (#9785)
2023-11-17 04:29:08 +08:00
random.cu
Implement column sampler in CUDA. (#9785)
2023-11-17 04:29:08 +08:00
random.h
Implement column sampler in CUDA. (#9785)
2023-11-17 04:29:08 +08:00
ranking_utils.cc
Rework the precision metric. (#9222)
2023-06-02 20:49:43 +08:00
ranking_utils.cu
Use the new DeviceOrd in the linalg module. (#9527)
2023-08-29 13:37:29 +08:00
ranking_utils.cuh
Implement NDCG cache. (#8893)
2023-03-13 22:16:31 +08:00
ranking_utils.h
Reorder if-else statements to allow using of cpu branches for sycl-devices (#9682)
2023-10-18 10:55:33 +08:00
ref_resource_view.h
Fix build for AppleClang 11 (#9684) (#9693)
2023-10-18 12:27:21 -07:00
row_set.h
Restore clang tidy test. (#8861)
2023-03-03 13:47:04 -08:00
stats.cc
Reorder if-else statements to allow using of cpu branches for sycl-devices (#9682)
2023-10-18 10:55:33 +08:00
stats.cu
Remove internal use of gpu_id. (#9568)
2023-09-20 23:29:51 +08:00
stats.cuh
Remove internal use of gpu_id. (#9568)
2023-09-20 23:29:51 +08:00
stats.h
Restore clang tidy test. (#8861)
2023-03-03 13:47:04 -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
Add support for cgroupv2. (#9651)
2023-10-12 09:36:36 +08:00
threading_utils.cuh
Define a new ranking parameter. (#8887)
2023-03-09 17:46:24 +08:00
threading_utils.h
Add support for cgroupv2. (#9651)
2023-10-12 09:36:36 +08:00
timer.cc
[Breaking] Switch from rabit to the collective communicator (#8257)
2022-10-05 14:39:01 -08:00
timer.h
[coll] Improvements and fixes for tracker and allreduce. (#9745)
2023-11-02 04:06:46 +08:00
transform_iterator.h
Fix IndexTransformIter. (#9155)
2023-05-12 21:25:54 +08:00
transform.h
Remove internal use of gpu_id. (#9568)
2023-09-20 23:29:51 +08:00
type.h
Serialize expand entry for allgather. (#9702)
2023-10-24 14:33:28 +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