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

@@ -39,7 +39,7 @@ A CUDA capable GPU with at least compute capability >= 3.5 (the algorithm depend
Building the plug-in requires CUDA Toolkit 7.5 or later.
The plugin also depends on CUB 1.5.2 - https://github.com/NVlabs/cub/tree/1.5.2
The plugin also depends on CUB 1.6.4 - https://nvlabs.github.io/cub/
CUB is a header only cuda library which provides sort/reduce/scan primitives.
@@ -70,6 +70,8 @@ The build process generates an xgboost library and executable as normal but cont
## Author
Rory Mitchell
Report any bugs to r.a.mitchell.nz at google mail.
Please report bugs to the xgboost/issues page. You can tag me with @RAMitchell.
Otherwise I can be contacted at r.a.mitchell.nz at gmail.

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;