Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Accept string for ArrayInterface constructor. (#5799)

Browse Source
This commit is contained in:
Jiaming Yuan
2020-06-27 00:06:54 +08:00
committed by GitHub
parent 95f11ed27e
commit 47c89775d6
4 changed files with 92 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/cpp/helpers.cc
View File

@@ -182,7 +182,13 @@ Json RandomDataGenerator::ArrayInterfaceImpl(HostDeviceVector<float> *storage,
this->GenerateDense(storage);
Json array_interface {Object()};
array_interface["data"] = std::vector<Json>(2);
array_interface["data"][0] = Integer(reinterpret_cast<int64_t>(storage->DevicePointer()));
if (storage->DeviceCanRead()) {
array_interface["data"][0] =
Integer(reinterpret_cast<int64_t>(storage->ConstDevicePointer()));
} else {
array_interface["data"][0] =
Integer(reinterpret_cast<int64_t>(storage->ConstHostPointer()));
}
array_interface["data"][1] = Boolean(false);
array_interface["shape"] = std::vector<Json>(2);