File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed
Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -110,7 +110,7 @@ class EvaluateSplitAgent {
110110 return gpair;
111111 }
112112
113- __device__ __forceinline__ void Numerical (DeviceSplitCandidate *__restrict__ best_split) {
113+ __device__ __forceinline__ void Numerical (DeviceSplitCandidate *best_split) {
114114 for (int scan_begin = gidx_begin; scan_begin < gidx_end; scan_begin += kBlockSize ) {
115115 bool thread_active = (scan_begin + threadIdx .x ) < gidx_end;
116116 GradientPairInt64 bin = thread_active ? LoadGpair (node_histogram + scan_begin + threadIdx .x )
@@ -147,7 +147,7 @@ class EvaluateSplitAgent {
147147 }
148148 }
149149
150- __device__ __forceinline__ void OneHot (DeviceSplitCandidate *__restrict__ best_split) {
150+ __device__ __forceinline__ void OneHot (DeviceSplitCandidate *best_split) {
151151 for (int scan_begin = gidx_begin; scan_begin < gidx_end; scan_begin += kBlockSize ) {
152152 bool thread_active = (scan_begin + threadIdx .x ) < gidx_end;
153153
Original file line number Diff line number Diff line change @@ -114,7 +114,9 @@ TEST(TestVirtualMem, Version) {
114114#if defined(xgboost_IS_WIN)
115115 ASSERT_FALSE (pinned.IsVm ());
116116#else // defined(xgboost_IS_WIN)
117- if (major >= 12 && minor >= 5 ) {
117+ if (major > 12 ) {
118+ ASSERT_TRUE (pinned.IsVm ());
119+ } else if (major >= 12 && minor >= 5 ) {
118120 ASSERT_TRUE (pinned.IsVm ());
119121 } else {
120122 ASSERT_FALSE (pinned.IsVm ());
You can’t perform that action at this time.
0 commit comments