GPU Plugin: Add bosch demo, update build instructions (#1872)

This commit is contained in:
Rory Mitchell
2016-12-15 19:57:27 +13:00
committed by Tianqi Chen
parent edc356f7ec
commit d943720883
5 changed files with 88 additions and 4 deletions

View File

@@ -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));

View File

@@ -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;