Initialization.

dmlc · Sep 25, 2020 · 7930390 · 7930390
1 parent 8c01055
commit 7930390
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/tree/updater_gpu_hist.cu b/src/tree/updater_gpu_hist.cu
@@ -602,7 +602,7 @@ struct GPUHistMakerDevice {
     auto is_cat = candidate.split.is_cat;
     if (is_cat) {
       auto cat = common::AsCat(candidate.split.fvalue);
-      std::vector<uint32_t> split_cats(LBitField32::ComputeStorageSize(std::max(cat+1, 1)));
+      std::vector<uint32_t> split_cats(LBitField32::ComputeStorageSize(std::max(cat+1, 1)), 0);
       LBitField32 cats_bits(split_cats);
       cats_bits.Set(cat);
       dh::CopyToD(split_cats, &node_categories);

diff --git a/tests/cpp/tree/test_gpu_hist.cu b/tests/cpp/tree/test_gpu_hist.cu
@@ -133,6 +133,7 @@ TEST(GpuHist, BuildHistSharedMem) {
 TEST(GpuHist, ApplySplit) {
   RegTree tree;
   ExpandEntry candidate;
+  candidate.nid = 0;
   candidate.left_weight = 1.0f;
   candidate.right_weight = 2.0f;
   candidate.base_weight = 3.0f;