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

sync up May15 2023

Browse Source
This commit is contained in:
amdsc21
2023-05-15 18:59:18 +02:00
parent b066accad6 7375bd058b
commit 8cad8c693c
37 changed files with 628 additions and 398 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/cpp/data/test_array_interface.cu
View File

@@ -1,5 +1,5 @@
/*!
* Copyright 2021 by Contributors
/**
* Copyright 2021-2023, XGBoost Contributors
*/
#include <gtest/gtest.h>
#include <xgboost/host_device_vector.h>
@@ -22,31 +22,19 @@ TEST(ArrayInterface, Stream) {
HostDeviceVector<float> storage;
auto arr_str = RandomDataGenerator{kRows, kCols, 0}.GenerateArrayInterface(&storage);
#if defined(XGBOOST_USE_CUDA)
cudaStream_t stream;
cudaStreamCreate(&stream);
#elif defined(XGBOOST_USE_HIP)
hipStream_t stream;
hipStreamCreate(&stream);
#endif
dh::CUDAStream stream;
auto j_arr =Json::Load(StringView{arr_str});
j_arr["stream"] = Integer(reinterpret_cast<int64_t>(stream));
auto j_arr = Json::Load(StringView{arr_str});
j_arr["stream"] = Integer(reinterpret_cast<int64_t>(stream.Handle()));
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);
std::uint64_t dur = 1e9;
dh::LaunchKernel{1, 1, 0, stream.View()}(SleepForTest, out.data().get(), dur);
ArrayInterface<2> arr(arr_str);
auto t = out[0];
CHECK_GE(t, dur);
#if defined(XGBOOST_USE_CUDA)
cudaStreamDestroy(stream);
#elif defined(XGBOOST_USE_HIP)
hipStreamDestroy(stream);
#endif
}
TEST(ArrayInterface, Ptr) {