 Rong Ou
|
a9ec2dd295
|
only copy the model once when predicting multiple batches (#4457)
|
2019-05-15 11:04:22 +12:00 |
|
|
Rong Ou
|
df2cdaca50
|
add cuda 10.1 support (#4468)
|
2019-05-14 18:30:58 +00:00 |
|
|
Rong Ou
|
be0f346ec9
|
mgpu predictor using explicit offsets (#4438)
* mgpu prediction using explicit sharding
|
2019-05-11 09:35:06 +12:00 |
|
|
Rong Ou
|
feb6ae3e18
|
Initial support for external memory in gpu_predictor (#4284)
|
2019-05-03 13:01:27 +12:00 |
|
|
Rong Ou
|
eaab364a63
|
More explict sharding methods for device memory (#4396)
* Rename the Reshard method to Shard
* Add a new Reshard method for sharding a vector that's already sharded
|
2019-05-01 11:47:22 +12:00 |
|
|
Rong Ou
|
2c61f02add
|
fix broken python test (#4395)
|
2019-04-23 16:01:23 -07:00 |
|
|
Rong Ou
|
f4521bf6aa
|
refactor tests to get rid of duplication (#4358)
* refactor tests to get rid of duplication
* address review comments
|
2019-04-12 00:21:48 -07:00 |
|
|
Rong Ou
|
81c1cd40ca
|
add a test for cpu predictor using external memory (#4308)
* add a test for cpu predictor using external memory
* allow different page size for testing
|
2019-04-10 13:25:10 +12:00 |
|
|
Rong Ou
|
7ea5b772fb
|
do not filter shared library files (#4303)
|
2019-03-28 19:40:54 +08:00 |
|
|
Rong Ou
|
8c8021dfa7
|
use all cores to build on linux (#4304)
|
2019-03-27 19:51:08 -07:00 |
|
|
Rong Ou
|
5aa42b5f11
|
jenkins build for cuda 10.0 (#4281)
* jenkins build for cuda 10.0
* yum install nccl2 for cuda 10.0
|
2019-03-22 22:35:18 -07:00 |
|
|
Rong Ou
|
9837b09b20
|
support cuda 10.1 (#4223)
* support cuda 10.1
* add cuda 10.1 to jenkins build matrix
|
2019-03-08 12:22:12 +13:00 |
|
|
Rong Ou
|
0944360416
|
minor fix: log InitDataOnce() only when it is actually called (#4206)
|
2019-03-08 10:53:09 +13:00 |
|
|
Rong Ou
|
8e0a08fbcf
|
Update python benchmarking script (#4164)
* a few tweaks to speed up data generation
* del variable to save memory
* switch to random numpy arrays
|
2019-02-21 15:16:09 +13:00 |
|
|
Rong Ou
|
d506a8bc63
|
[jvm-packages] add verbosity param (#4138)
|
2019-02-13 20:57:17 -08:00 |
|
|
Rong Ou
|
3be1b9ae30
|
reformat benchmark_tree.py to get rid of lint errors (#4126)
|
2019-02-13 18:54:56 +13:00 |
|
|
Rong Ou
|
9b917cda4f
|
[jvm-packages] fix simple logic error :) (#4128)
@CodingCat
|
2019-02-11 21:47:30 -08:00 |
|
|
Rong Ou
|
2a9b085bc8
|
[jvm-packages] minor fix of params (#4114)
|
2019-02-08 00:21:59 -08:00 |
|