@@ -105,7 +105,7 @@ BatchSet<SparsePage> SparsePageDMatrix::GetRowBatches() {
105
105
}
106
106
107
107
BatchSet<CSCPage> SparsePageDMatrix::GetColumnBatches (Context const *ctx) {
108
- auto id = MakeCache (this , " .col.page" , on_host_ , cache_prefix_, &cache_info_);
108
+ auto id = MakeCache (this , " .col.page" , false , cache_prefix_, &cache_info_);
109
109
CHECK_NE (this ->Info ().num_col_ , 0 );
110
110
this ->InitializeSparsePage (ctx);
111
111
if (!column_source_) {
@@ -119,7 +119,7 @@ BatchSet<CSCPage> SparsePageDMatrix::GetColumnBatches(Context const *ctx) {
119
119
}
120
120
121
121
BatchSet<SortedCSCPage> SparsePageDMatrix::GetSortedColumnBatches (Context const *ctx) {
122
- auto id = MakeCache (this , " .sorted.col.page" , on_host_ , cache_prefix_, &cache_info_);
122
+ auto id = MakeCache (this , " .sorted.col.page" , false , cache_prefix_, &cache_info_);
123
123
CHECK_NE (this ->Info ().num_col_ , 0 );
124
124
this ->InitializeSparsePage (ctx);
125
125
if (!sorted_column_source_) {
@@ -138,11 +138,11 @@ BatchSet<GHistIndexMatrix> SparsePageDMatrix::GetGradientIndex(Context const *ct
138
138
CHECK_GE (param.max_bin , 2 );
139
139
}
140
140
detail::CheckEmpty (batch_param_, param);
141
- auto id = MakeCache (this , " .gradient_index.page" , on_host_ , cache_prefix_, &cache_info_);
141
+ auto id = MakeCache (this , " .gradient_index.page" , false , cache_prefix_, &cache_info_);
142
142
if (!cache_info_.at (id)->written || detail::RegenGHist (batch_param_, param)) {
143
143
this ->InitializeSparsePage (ctx);
144
144
cache_info_.erase (id);
145
- id = MakeCache (this , " .gradient_index.page" , on_host_ , cache_prefix_, &cache_info_);
145
+ id = MakeCache (this , " .gradient_index.page" , false , cache_prefix_, &cache_info_);
146
146
LOG (INFO) << " Generating new Gradient Index." ;
147
147
// Use sorted sketch for approx.
148
148
auto sorted_sketch = param.regen ;
0 commit comments