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

Partial fix for CTK 12.5 (#10574)

Browse Source
This commit is contained in:
Jiaming Yuan 2024-07-16 17:41:50 +08:00 committed by GitHub
parent 370dce9d57
commit 5a92ffe3ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/tree/updater_gpu_common.cuh
View File

@ -1,18 +1,13 @@
/*!
* Copyright 2017-2019 XGBoost contributors
/**
* Copyright 2017-2024, XGBoost contributors
*/
#pragma once
#include <thrust/random.h>
#include <cstdio>
#include <cub/cub.cuh>
#include <stdexcept>
#include <string>
#include <vector>
#include "../common/categorical.h"
#include "../common/device_helpers.cuh"
#include "../common/random.h"
#include <limits> // for numeric_limits
#include <ostream> // for ostream
#include "gpu_hist/histogram.cuh"
#include "param.h"
#include "xgboost/base.h"
namespace xgboost::tree {
struct GPUTrainingParam {
@ -54,8 +49,8 @@ enum DefaultDirection {
};
struct DeviceSplitCandidate {
float loss_chg {-FLT_MAX};
DefaultDirection dir {kLeftDir};
float loss_chg{-std::numeric_limits<float>::max()};
DefaultDirection dir{kLeftDir};
int findex {-1};
float fvalue {0};
// categorical split, either it's the split category for OHE or the threshold for partition-based

1
src/tree/updater_gpu_hist.cu
View File

@ -19,6 +19,7 @@
#include "../common/cuda_context.cuh" // CUDAContext
#include "../common/device_helpers.cuh"
#include "../common/hist_util.h"
#include "../common/random.h" // for ColumnSampler, GlobalRandom
#include "../common/timer.h"
#include "../data/ellpack_page.cuh"
#include "../data/ellpack_page.h"