Fix prediction.

dmlc · Aug 25, 2020 · d86f917 · d86f917
1 parent 7e5f67e
commit d86f917
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/predictor/gpu_predictor.cu b/src/predictor/gpu_predictor.cu
@@ -196,6 +196,7 @@ __device__ float GetLeafWeight(bst_uint ridx, const RegTree::Node* tree,
         nidx = n.RightChild();
       }
     }
+    n = tree[nidx];
   }
   return tree[nidx].LeafValue();
 }