GPU Plugin: Add bosch demo, update build instructions (#1872)
This commit is contained in:
committed by
Tianqi Chen
parent
edc356f7ec
commit
d943720883
@@ -414,7 +414,6 @@ class bulk_allocator {
|
||||
}
|
||||
|
||||
_size = get_size_bytes(args...);
|
||||
std::cout << "trying to allocate: " << _size << "\n";
|
||||
|
||||
safe_cuda(cudaMalloc(&d_ptr, _size));
|
||||
|
||||
|
||||
@@ -311,7 +311,6 @@ float GPUBuilder::GetSubsamplingRate(MetaInfo info) {
|
||||
uint32_t max_nodes_level = 1 << param.max_depth;
|
||||
size_t required = 10 * info.num_row + 40 * info.num_nonzero
|
||||
+ 64 * max_nodes + 76 * max_nodes_level * info.num_col;
|
||||
std::cout << "required: " << required << "\n";
|
||||
size_t available = dh::available_memory();
|
||||
while (available < required) {
|
||||
subsample -= 0.05;
|
||||
|
||||
Reference in New Issue
Block a user