simplify resizing bit vectors

dmlc · Jul 1, 2023 · 6dbc9ac · 6dbc9ac
1 parent 3186816
commit 6dbc9ac
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/src/predictor/gpu_predictor.cu b/src/predictor/gpu_predictor.cu
@@ -803,20 +803,14 @@ class ColumnSplitHelper {
                                dh::caching_device_vector<BitType>* missing_storage,
                                std::size_t total_bits) const {
     auto const size = BitVector::ComputeStorageSize(total_bits);
-    auto const old_decision_size = decision_storage->size();
-    if (old_decision_size < size) {
+    if (decision_storage->size() < size) {
       decision_storage->resize(size);
     }
-    if (old_decision_size != 0) {
-      thrust::fill(ctx_->CUDACtx()->CTP(), decision_storage->begin(), decision_storage->end(), 0);
-    }
-    auto const old_missing_size = missing_storage->size();
-    if (old_missing_size < size) {
+    thrust::fill(ctx_->CUDACtx()->CTP(), decision_storage->begin(), decision_storage->end(), 0);
+    if (missing_storage->size() < size) {
       missing_storage->resize(size);
     }
-    if (old_missing_size != 0) {
-      thrust::fill(ctx_->CUDACtx()->CTP(), missing_storage->begin(), missing_storage->end(), 0);
-    }
+    thrust::fill(ctx_->CUDACtx()->CTP(), missing_storage->begin(), missing_storage->end(), 0);
   }
 
   Context const* ctx_;