Skip to content

Commit

Permalink
[backport] Partial fix for CTK 12.5 (dmlc#10574)
Browse files Browse the repository at this point in the history
  • Loading branch information
trivialfis committed Jul 25, 2024
1 parent 3fd8451 commit 8e8341a
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 13 deletions.
21 changes: 8 additions & 13 deletions src/tree/updater_gpu_common.cuh
Original file line number Diff line number Diff line change
@@ -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 {
Expand Down Expand Up @@ -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
Expand Down
1 change: 1 addition & 0 deletions src/tree/updater_gpu_hist.cu
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down

0 comments on commit 8e8341a

Please sign in to comment.