diff --git a/gputreeshap b/gputreeshap index 04410099299e..1de23c95ff07 160000 --- a/gputreeshap +++ b/gputreeshap @@ -1 +1 @@ -Subproject commit 04410099299ec918c75d00e385da35cf2e5aa87c +Subproject commit 1de23c95ff07d086db02837fb4a746b6924abbd5 diff --git a/src/predictor/gpu_predictor.cu b/src/predictor/gpu_predictor.cu index 12fb16828d05..65d90982b2e5 100644 --- a/src/predictor/gpu_predictor.cu +++ b/src/predictor/gpu_predictor.cu @@ -256,10 +256,10 @@ class DeviceModel { auto& src_stats = model.trees.at(tree_idx)->GetStats(); dh::safe_cuda(cudaMemcpyAsync( d_nodes + h_tree_segments[tree_idx - tree_begin], src_nodes.data(), - sizeof(RegTree::Node) * src_nodes.size(),cudaMemcpyDefault)); + sizeof(RegTree::Node) * src_nodes.size(), cudaMemcpyDefault)); dh::safe_cuda(cudaMemcpyAsync( d_stats + h_tree_segments[tree_idx - tree_begin], src_stats.data(), - sizeof(RTreeNodeStat) * src_stats.size(),cudaMemcpyDefault)); + sizeof(RTreeNodeStat) * src_stats.size(), cudaMemcpyDefault)); } tree_group = std::move(HostDeviceVector(model.tree_info.size(), 0, gpu_id));