55ee272ea8
* Extend array interface to handle ndarray. The `ArrayInterface` class is extended to support multi-dim array inputs. Previously this class handles only 2-dim (vector is also matrix). This PR specifies the expected dimension at compile-time and the array interface can perform various checks automatically for input data. Also, adapters like CSR are more rigorous about their input. Lastly, row vector and column vector are handled without intervention from the caller.
56 lines
1.4 KiB
Plaintext
56 lines
1.4 KiB
Plaintext
/*!
|
|
* Copyright 2021 by Contributors
|
|
*/
|
|
#include <gtest/gtest.h>
|
|
#include <xgboost/host_device_vector.h>
|
|
#include "../helpers.h"
|
|
#include "../../../src/data/array_interface.h"
|
|
|
|
namespace xgboost {
|
|
|
|
__global__ void SleepForTest(uint64_t *out, uint64_t duration) {
|
|
auto start = clock64();
|
|
auto t = 0;
|
|
while (t < duration) {
|
|
t = clock64() - start;
|
|
}
|
|
out[0] = t;
|
|
}
|
|
|
|
TEST(ArrayInterface, Stream) {
|
|
size_t constexpr kRows = 10, kCols = 10;
|
|
HostDeviceVector<float> storage;
|
|
auto arr_str = RandomDataGenerator{kRows, kCols, 0}.GenerateArrayInterface(&storage);
|
|
|
|
cudaStream_t stream;
|
|
cudaStreamCreate(&stream);
|
|
|
|
auto j_arr =Json::Load(StringView{arr_str});
|
|
j_arr["stream"] = Integer(reinterpret_cast<int64_t>(stream));
|
|
Json::Dump(j_arr, &arr_str);
|
|
|
|
dh::caching_device_vector<uint64_t> out(1, 0);
|
|
uint64_t dur = 1e9;
|
|
dh::LaunchKernel{1, 1, 0, stream}(SleepForTest, out.data().get(), dur);
|
|
ArrayInterface<2> arr(arr_str);
|
|
|
|
auto t = out[0];
|
|
CHECK_GE(t, dur);
|
|
|
|
cudaStreamDestroy(stream);
|
|
}
|
|
|
|
TEST(ArrayInterface, Ptr) {
|
|
std::vector<float> h_data(10);
|
|
ASSERT_FALSE(ArrayInterfaceHandler::IsCudaPtr(h_data.data()));
|
|
dh::safe_cuda(cudaGetLastError());
|
|
|
|
dh::device_vector<float> d_data(10);
|
|
ASSERT_TRUE(ArrayInterfaceHandler::IsCudaPtr(d_data.data().get()));
|
|
dh::safe_cuda(cudaGetLastError());
|
|
|
|
ASSERT_FALSE(ArrayInterfaceHandler::IsCudaPtr(nullptr));
|
|
dh::safe_cuda(cudaGetLastError());
|
|
}
|
|
} // namespace xgboost
|