Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

macro format

Browse Source
This commit is contained in:
amdsc21 2023-03-11 06:57:24 +01:00
parent b0dacc5a80
commit 7d96758382
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/context.cc
View File

@ -47,7 +47,7 @@ void Context::ConfigureGpuId(bool require_gpu) {
// Just set it to CPU, don't think about it. // Just set it to CPU, don't think about it.
this->UpdateAllowUnknown(Args{{"gpu_id", std::to_string(kCpuId)}}); this->UpdateAllowUnknown(Args{{"gpu_id", std::to_string(kCpuId)}});
(void)(require_gpu); (void)(require_gpu);
#endif // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_ #endif // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
common::SetDevice(this->gpu_id); common::SetDevice(this->gpu_id);
} }

2
src/tree/fit_stump.cc
View File

@ -61,7 +61,7 @@ inline void FitStump(Context const*, linalg::TensorView<GradientPair const, 2>,
linalg::VectorView<float>) { linalg::VectorView<float>) {
common::AssertGPUSupport(); common::AssertGPUSupport();
} }
#endif // !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_C #endif // !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
} // namespace cuda_impl } // namespace cuda_impl
void FitStump(Context const* ctx, HostDeviceVector<GradientPair> const& gpair, void FitStump(Context const* ctx, HostDeviceVector<GradientPair> const& gpair,

2
src/tree/gpu_hist/row_partitioner.cuh
View File

@ -124,7 +124,7 @@ void SortPositionBatch(common::Span<const PerNodeData<OpDataT>> d_batch_info,
dh::device_vector<int8_t>* tmp, dh::device_vector<int8_t>* tmp,
#if defined(XGBOOST_USE_HIP) #if defined(XGBOOST_USE_HIP)
hipStream_t stream hipStream_t stream
#else #elif defined(XGBOOST_USE_CUDA)
cudaStream_t stream cudaStream_t stream
#endif #endif
) { ) {