3e26107a9c
* Rename `GenericParameter` to `Context`. * Rename header file to reflect the change. * Rename all references.
26 lines
856 B
Plaintext
26 lines
856 B
Plaintext
/*!
|
|
* Copyright 2022 by XGBoost Contributors
|
|
*/
|
|
#include <thrust/execution_policy.h>
|
|
#include <thrust/functional.h> // thrust:plus
|
|
|
|
#include "device_helpers.cuh" // dh::Reduce, safe_cuda, dh::XGBCachingDeviceAllocator
|
|
#include "numeric.h"
|
|
#include "xgboost/context.h" // Context
|
|
#include "xgboost/host_device_vector.h" // HostDeviceVector
|
|
|
|
namespace xgboost {
|
|
namespace common {
|
|
namespace cuda {
|
|
double Reduce(Context const* ctx, HostDeviceVector<float> const& values) {
|
|
values.SetDevice(ctx->gpu_id);
|
|
auto const d_values = values.ConstDeviceSpan();
|
|
dh::XGBCachingDeviceAllocator<char> alloc;
|
|
auto res = dh::Reduce(thrust::cuda::par(alloc), d_values.data(),
|
|
d_values.data() + d_values.size(), 0.0, thrust::plus<double>{});
|
|
return res;
|
|
}
|
|
} // namespace cuda
|
|
} // namespace common
|
|
} // namespace xgboost
|